Home
last modified time | relevance | path

Searched refs:TestMatrixTensorHandle (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_test.cc65 TFE_TensorHandle* m = TestMatrixTensorHandle(ctx); in BM_InitOp()
87 TFE_TensorHandle* m = TestMatrixTensorHandle(ctx); in BM_Execute()
126 TFE_TensorHandle* m = TestMatrixTensorHandle(ctx); in BM_Execute_Identity()
182 TFE_TensorHandle* h = TestMatrixTensorHandle(ctx); in TEST()
207 TFE_TensorHandle* hcpu = TestMatrixTensorHandle(ctx); in TensorHandleCopyBetweenDevices()
283 TFE_TensorHandle* hcpu = TestMatrixTensorHandle(ctx); in TensorHandleCopyBetweenDevicesError()
324 TFE_TensorHandle* hcpu = TestMatrixTensorHandle(ctx); in TensorHandleCopyBetweenTwoGPUDevices()
410 TFE_TensorHandle* hcpu = TestMatrixTensorHandle(ctx); in TensorHandleSilentCopy()
483 TFE_TensorHandle* m = TestMatrixTensorHandle(ctx); in SetAndGetOpDevices()
554 TFE_TensorHandle* hcpu = TestMatrixTensorHandle(ctx); in TEST()
[all …]
H A Dc_api_remote_test_util.cc104 TFE_TensorHandle* h0_task0 = TestMatrixTensorHandle(ctx); in TestRemoteExecuteSilentCopies()
105 TFE_TensorHandle* h1_task0 = TestMatrixTensorHandle(ctx); in TestRemoteExecuteSilentCopies()
111 handles_task0.push_back(TestMatrixTensorHandle(ctx)); in TestRemoteExecuteSilentCopies()
H A Dc_api_remote_test.cc66 TFE_TensorHandle* h0_task0 = TestMatrixTensorHandle(ctx); in TestRemoteExecute()
67 TFE_TensorHandle* h1_task0 = TestMatrixTensorHandle(ctx); in TestRemoteExecute()
H A Dc_api_experimental_test.cc333 TFE_TensorHandle* m = TestMatrixTensorHandle(ctx); in Executor_MatMul_CPU()
378 TFE_TensorHandle* m = TestMatrixTensorHandle(ctx); in TEST()
443 TFE_TensorHandle* hcpu = TestMatrixTensorHandle(ctx); in TEST()
495 TFE_TensorHandle* h_default = TestMatrixTensorHandle(ctx); in TEST()
H A Dc_api_cluster_test.cc61 TFE_TensorHandle* h0_task0 = TestMatrixTensorHandle(ctx); in CheckRemoteMatMulExecutesOK()
160 TFE_TensorHandle* h0_task0_new = TestMatrixTensorHandle(ctx); in TestRemoteExecuteChangeServerDef()
H A Dc_api_test_util.h43 TFE_TensorHandle* TestMatrixTensorHandle(TFE_Context* ctx);
H A Dcustom_device_test.cc42 TFE_TensorHandle* hcpu = TestMatrixTensorHandle(context); in TEST()
281 TestMatrixTensorHandle(context.get()), TFE_DeleteTensorHandle); in TEST()
H A Dc_api_test_util.cc97 TFE_TensorHandle* TestMatrixTensorHandle(TFE_Context* ctx) { in TestMatrixTensorHandle() function
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/
H A Dc_api_multi_client_test.cc98 TFE_TensorHandle* in = TestMatrixTensorHandle(ctx); in TEST()
186 TFE_TensorHandle* in = TestMatrixTensorHandle(ctx); in TEST()
H A Dc_api_coordination_test.cc154 TFE_TensorHandle* in = TestMatrixTensorHandle(ctx); in TEST()
379 TFE_TensorHandle* in = TestMatrixTensorHandle(ctx); in TEST()