Home
last modified time | relevance | path

Searched refs:TFE_OpAddInput (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.cc139 TFE_OpAddInput(op, a, status); in MatMulOp()
141 TFE_OpAddInput(op, b, status); in MatMulOp()
154 TFE_OpAddInput(op, a, status); in IdentityOp()
167 TFE_OpAddInput(op, a, status); in ShapeOp()
195 TFE_OpAddInput(op, input, status); in MinOp()
197 TFE_OpAddInput(op, axis, status); in MinOp()
Dc_api_test.cc948 TFE_OpAddInput(op, m, status); in FunctionDefAndExecute()
995 TFE_OpAddInput(matmul, m, status); in BM_ExecuteFunction()
1042 TFE_OpAddInput(op, var_handle, status); in CreateVariable()
1054 TFE_OpAddInput(op, value_handle.get(), status); in CreateVariable()
1081 TFE_OpAddInput(op, var_handle, status); in TEST()
1122 TFE_OpAddInput(op, var_handle, status); in BM_ReadVariable()
1167 TFE_OpAddInput(op, tensor_handle, status); in TEST()
1242 TFE_OpAddInput(minOp, input, status); in TEST()
1244 TFE_OpAddInput(minOp, axis, status); in TEST()
1282 TFE_OpAddInput(concatOp, dim, status); in TEST()
[all …]
Dc_api.h267 TF_CAPI_EXPORT extern void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input,
Dc_api_experimental_test.cc343 TFE_OpAddInput(op, h, status); in TEST()
419 TFE_OpAddInput(op, h, status); in TEST()
Dc_api.cc1239 void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input, TF_Status* status) { in TFE_OpAddInput() function
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Deval_util.cc133 TFE_OpAddInput(op, input_handle, status); in EvaluateOperation()
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc387 TFE_OpAddInput(op, queue, status); in createTFEEnqueue()
389 TFE_OpAddInput(op, tensor, status); in createTFEEnqueue()
410 TFE_OpAddInput(op, queue, status); in createTFEDequeue()
Dc_api_experimental_test.cc142 TFE_OpAddInput(send_op, m, status); in TEST()
/external/tensorflow/tensorflow/java/src/main/native/
Deager_operation_builder_jni.cc135 TFE_OpAddInput(op, tensor_handle, status); in Java_org_tensorflow_EagerOperationBuilder_addInput()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc214 TFE_OpAddInput(op, handle, out_status); in EagerCast() local
Dpywrap_tfe_src.cc847 TFE_OpAddInput(op, inputs->at(i), out_status); in TFE_Py_ExecuteCancelable()
3250 TFE_OpAddInput(op, EagerTensor_Handle(handle.get()), status);
3371 TFE_OpAddInput(op, input_handle, status);
3697 TFE_OpAddInput(op, input_handle, status);