| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test_util.cc | 221 TFE_OpAddInput(op, var_handle, status); in TestVariable()
233 TFE_OpAddInput(op, value_handle.get(), status); in TestVariable()
252 TFE_OpAddInput(op, a, status); in AddOp()
254 TFE_OpAddInput(op, b, status); in AddOp()
267 TFE_OpAddInput(op, a, status); in MatMulOp()
269 TFE_OpAddInput(op, b, status); in MatMulOp()
282 TFE_OpAddInput(op, a, status); in IdentityOp()
295 TFE_OpAddInput(op, a, status); in ShapeOp()
323 TFE_OpAddInput(op, input, status); in MinOp()
325 TFE_OpAddInput(op, axis, status); in MinOp()
|
| D | custom_device_test.cc | 136 TFE_OpAddInput(op.get(), var_handle, status.get()); in TEST()
137 TFE_OpAddInput(op.get(), one.get(), status.get()); in TEST()
148 TFE_OpAddInput(op.get(), var_handle, status.get()); in TEST()
174 TFE_OpAddInput(op.get(), var_handle, status.get()); in TEST()
221 TFE_OpAddInput(op.get(), var_handle, status.get()); in TEST()
222 TFE_OpAddInput(op.get(), one.get(), status.get()); in TEST()
234 TFE_OpAddInput(op.get(), var_handle, status.get()); in TEST()
250 TFE_OpAddInput(op.get(), var_handle, status.get()); in TEST()
|
| D | c_api_test.cc | 93 TFE_OpAddInput(matmul, m, status); in BM_Execute()
95 TFE_OpAddInput(matmul, m, status); in BM_Execute()
134 TFE_OpAddInput(identity, m, status); in BM_Execute_Identity()
1026 TFE_OpAddInput(op, m, status); in FunctionDefAndExecute()
1097 TFE_OpAddInput(op, m, status); in RunAddFunction()
1177 TFE_OpAddInput(matmul, m, status); in BM_ExecuteFunction()
1215 TFE_OpAddInput(op, var_handle, status); in TEST()
1260 TFE_OpAddInput(op, var_handle, status); in BM_ReadVariable()
1301 TFE_OpAddInput(op, tensor_handle, status); in TEST()
1360 TFE_OpAddInput(op, tensor_handle, status); in TEST()
[all …]
|
| D | c_api_remote_test_util.cc | 147 TFE_OpAddInput(matmul, h0_task0, status); in TestRemoteExecuteSilentCopies()
149 TFE_OpAddInput(matmul, has_packed_input ? packed_handle : h1_task2, status); in TestRemoteExecuteSilentCopies()
|
| D | custom_device_testutil.cc | 125 TFE_OpAddInput(op, t->tensor, s); in LoggingDeviceExecute()
127 TFE_OpAddInput(op, input, s); in LoggingDeviceExecute()
|
| D | c_api_distributed_test.cc | 198 TFE_OpAddInput(func, packed_handle, status); in TestFunctionWithPackedInput()
389 TFE_OpAddInput(func, var_handle, status); in TEST()
513 TFE_OpAddInput(func, var_handle, status); in TestDistributedFunctionCancellation()
|
| D | c_api_cluster_test.cc | 98 TFE_OpAddInput(op, var, status); in ReadVariable()
306 TFE_OpAddInput(op, var_handle0, status); in TestRemoteExecuteUpdateServerDefResourceAccess()
319 TFE_OpAddInput(op, var_handle1, status); in TestRemoteExecuteUpdateServerDefResourceAccess()
|
| D | c_api.h | 267 TF_CAPI_EXPORT extern void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input,
|
| D | c_api_experimental_test.cc | 289 TFE_OpAddInput(op, h, status); in TEST()
|
| D | c_api.cc | 672 void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input, TF_Status* status) { in TFE_OpAddInput() function
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_testlib.cc | 65 TFE_OpAddInput(op.get(), handle_, status); in Destroy()
82 TFE_OpAddInput(op.get(), handle_, status); in Read()
102 TFE_OpAddInput(op.get(), handle_, status); in GeneralAssignment()
104 TFE_OpAddInput(op.get(), value, status); in GeneralAssignment()
166 TFE_OpAddInput(op.get(), input, status); in ExtractPerDeviceValues()
187 TFE_OpAddInput(op.get(), first, status); in Multiply()
189 TFE_OpAddInput(op.get(), second, status); in Multiply()
|
| D | parallel_device_testlib.h | 126 TFE_OpAddInput(op.get(), components[i], status); in CreatePerDeviceValues()
|
| D | parallel_device_test.cc | 370 TFE_OpAddInput(op.get(), value_one.get(), status.get()); in TEST()
405 TFE_OpAddInput(op.get(), input, status); in CollectiveSum()
555 TFE_OpAddInput(op.get(), parallel_value.get(), status.get()); in TEST()
|
| D | parallel_device_lib.cc | 239 TFE_OpAddInput(op_.get(), inputs[input_index], status); in Execute()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | eval_util.cc | 117 TFE_OpAddInput(op, input_handle, status); in EvaluateOperation()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | eager_operation_builder_jni.cc | 135 TFE_OpAddInput(op, tensor_handle, status); in Java_org_tensorflow_EagerOperationBuilder_addInput()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tensor.cc | 245 TFE_OpAddInput(op, handle, out_status); in EagerCast() local
|
| D | pywrap_tfe_src.cc | 873 TFE_OpAddInput(op, inputs->at(i), out_status); in TFE_Py_ExecuteCancelable()
3334 TFE_OpAddInput(op, EagerTensor_Handle(handle.get()), status);
3455 TFE_OpAddInput(op, input_handle, status);
3799 TFE_OpAddInput(op, input_handle, status);
|