Home
last modified time | relevance | path

Searched refs:eager_ctx (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dunified_api_testutil.cc137 TFE_Context* eager_ctx = in TestScalarTensorHandle() local
140 TFE_TensorHandle* input_eager = TestScalarTensorHandle(eager_ctx, value); in TestScalarTensorHandle()
151 TFE_Context* eager_ctx = in TestTensorHandleWithDimsFloat() local
155 TestTensorHandleWithDimsFloat(eager_ctx, data, dims, num_dims); in TestTensorHandleWithDimsFloat()
166 TFE_Context* eager_ctx = in TestTensorHandleWithDimsInt() local
170 TestTensorHandleWithDimsInt(eager_ctx, data, dims, num_dims); in TestTensorHandleWithDimsInt()
Dc_api_unified_experimental_test.cc64 TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(ctx, status.get()); in TEST_P() local
66 TFE_TensorHandle* t = TestScalarTensorHandle(eager_ctx, 2.0f); in TEST_P()
127 TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(ctx, status.get()); in TEST_P() local
129 TestMatrixTensorHandleWithInput(eager_ctx, vals, dims, num_dims); in TEST_P()
200 TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(ctx, status.get()); in TEST_P() local
202 TestMatrixTensorHandleWithInput(eager_ctx, vals1, dims, num_dims); in TEST_P()
211 TestMatrixTensorHandleWithInput(eager_ctx, vals2, dims, num_dims); in TEST_P()
286 TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(ctx, status.get()); in TEST_P() local
288 TestMatrixTensorHandleWithInput(eager_ctx, vals1, dims, num_dims); in TEST_P()
297 TestMatrixTensorHandleWithInput(eager_ctx, vals2, dims, num_dims); in TEST_P()
[all …]
Dgradients_util.cc87 TFE_Context* eager_ctx = in ScalarTensorHandle() local
91 TF_RETURN_IF_ERROR(ScalarTensorHandleHelper(eager_ctx, value, &input_eager)); in ScalarTensorHandle()
103 TFE_Context* eager_ctx = in TensorHandleWithDimsFloat() local
107 TF_RETURN_IF_ERROR(TensorHandleWithDimsFloatHelper(eager_ctx, data, dims, in TensorHandleWithDimsFloat()
119 TFE_Context* eager_ctx = in TensorHandleWithDimsInt() local
123 TF_RETURN_IF_ERROR(TensorHandleWithDimsIntHelper(eager_ctx, data, dims, in TensorHandleWithDimsInt()
/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
Dvariable.cc87 EagerContext* eager_ctx = reinterpret_cast<EagerContext*>(ctx); in CreateUninitialized() local
102 std::move(handles), eager_ctx, &packed_handle)); in CreateUninitialized()