Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

initial test for matmul with abstract tensors #40851

Merged
merged 18 commits into from
Jul 16, 2020
Merged

initial test for matmul with abstract tensors #40851

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
3525264
initial test for matmul with abstract tensors
Jun 26, 2020
05ea0a6
Fixed MatMul Abstract Tensor test, now passing
Jun 29, 2020
91f5554
added non-trivial matmul example with abstract tensors
Jun 29, 2020
5c6557c
cleaned up syntax and style for matmul tests with abstract tensors
Jun 29, 2020
dc6a18b
resolving nit style issues
Jun 29, 2020
880b51a
fixed signature for util function to accept dimensions as a parameter
Jun 30, 2020
fc8f609
updated tests to take in dims as parameters, and added graph tracing …
Jun 30, 2020
1b4085d
fixed memory leaks and added matrix addition test
Jul 1, 2020
55f67d7
added robust tracing test for multi-output graph with adds and matmul
Jul 1, 2020
c2c3c36
Tests passing, checking for memory leaks
Jul 6, 2020
d1a654b
fixed nit by adding non-trivial tensors to multioutput graph example
Jul 7, 2020
f38c6c1
uncommented call to free graph_ctx
amturati Jul 7, 2020
aef53d7
deleted extraneous comment
amturati Jul 13, 2020
090f164
fixed style with clang format
Jul 13, 2020
a6a649a
resolving merge conflict, updated style
Jul 13, 2020
5d928f3
fixed nit with a space
amturati Jul 13, 2020
f469566
fixed license spacing issue
Jul 13, 2020
2097af0
resolving final nits and merge
Jul 13, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
12 changes: 12 additions & 0 deletions tensorflow/c/eager/c_api_test_util.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -88,6 +88,18 @@ TFE_TensorHandle* TestMatrixTensorHandle(TFE_Context* ctx) {
return th;
}

TFE_TensorHandle* TestMatrixTensorHandleWithInput(TFE_Context* ctx, float data[], int64_t dims [], int num_dims) {
TF_Status* status = TF_NewStatus();
TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0],
num_dims, status);
memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t));
TFE_TensorHandle* th = TFE_NewTensorHandleFromTensor(ctx, t, status);
CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
TF_DeleteTensor(t);
TF_DeleteStatus(status);
return th;
}

TFE_TensorHandle* TestMatrixTensorHandle100x100(TFE_Context* ctx) {
constexpr int64_t dims[] = {100, 100};
constexpr int num_elements = dims[0] * dims[1];
Expand Down
3 changes: 3 additions & 0 deletions tensorflow/c/eager/c_api_test_util.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,9 @@ TFE_TensorHandle* DoubleTestMatrixTensorHandle(TFE_Context* ctx);
// Return a tensor handle containing a 2x2 matrix of floats
TFE_TensorHandle* TestMatrixTensorHandle(TFE_Context* ctx);

//Return a tensor handle containing 2D matrix containing given data and dimensions
TFE_TensorHandle* TestMatrixTensorHandleWithInput(TFE_Context* ctx, float data[], int64_t dims [], int num_dims);

// Return a tensor handle containing a 100x100 matrix of floats
TFE_TensorHandle* TestMatrixTensorHandle100x100(TFE_Context* ctx);

Expand Down