Home
last modified time | relevance | path

Searched refs:TFE_NewOp (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.cc202 TFE_Op* op = TFE_NewOp(ctx, "VarHandleOp", status); in TestVariable()
221 op = TFE_NewOp(ctx, "AssignVariableOp", status); in TestVariable()
253 TFE_Op* op = TFE_NewOp(ctx, "AddV2", status); in AddOp()
268 TFE_Op* op = TFE_NewOp(ctx, "MatMul", status); in MatMulOp()
283 TFE_Op* op = TFE_NewOp(ctx, "Identity", status); in IdentityOp()
296 TFE_Op* op = TFE_NewOp(ctx, "Shape", status); in ShapeOp()
324 TFE_Op* op = TFE_NewOp(ctx, "Min", status); in MinOp()
341 TFE_Op* op = TFE_NewOp(ctx, "CollectiveReduce", status); in AllReduceOp()
365 TFE_Op* op = TFE_NewOp(ctx, op_name.c_str(), status); in SendOp()
386 TFE_Op* op = TFE_NewOp(ctx, op_name.c_str(), status); in RecvOp()
Dcustom_device_test.cc83 TFE_NewOp(context.get(), "Identity", status.get()), TFE_DeleteOp); in TEST()
114 TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp); in TEST()
134 op.reset(TFE_NewOp(context.get(), "AssignVariableOp", status.get())); in TEST()
147 op.reset(TFE_NewOp(context.get(), "ReadVariableOp", status.get())); in TEST()
173 op.reset(TFE_NewOp(context.get(), "DestroyResourceOp", status.get())); in TEST()
199 TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp); in TEST()
219 op.reset(TFE_NewOp(context.get(), "AssignVariableOp", status.get())); in TEST()
232 op.reset(TFE_NewOp(context.get(), "ReadVariableOp", status.get())); in TEST()
249 op.reset(TFE_NewOp(context.get(), "DestroyResourceOp", status.get())); in TEST()
Dc_api_test.cc88 TFE_Op* matmul = TFE_NewOp(ctx, "MatMul", status); in BM_Execute()
127 TFE_Op* identity = TFE_NewOp(ctx, "Identity", status); in BM_Execute_Identity()
1021 TFE_Op* non_comm_op = TFE_NewOp(ctx, "TestNonCommUnavailable", status); in TEST()
1029 TFE_Op* comm_op = TFE_NewOp(ctx, "TestCommUnavailable", status); in TEST()
1037 TFE_Op* fn1 = TFE_NewOp(ctx, "FunctionWithTestNonCommUnavailable", status); in TEST()
1045 TFE_Op* fn2 = TFE_NewOp(ctx, "FunctionWithTestCommUnavailable", status); in TEST()
1144 TFE_Op* op = TFE_NewOp(ctx, "MatMulFunction", status); in FunctionDefAndExecute()
1190 TFE_Op* op = TFE_NewOp(ctx, "AddFunction", status); in RunAddFunction()
1293 TFE_Op* matmul = TFE_NewOp(ctx, "MatMulFunction", status); in BM_ExecuteFunction()
1329 TFE_Op* op = TFE_NewOp(ctx, "ReadVariableOp", status); in TEST()
[all …]
Dc_api_distributed_test.cc196 TFE_Op* func = TFE_NewOp(ctx, "AddVariablesFunction", status); in TestFunctionWithPackedInput()
387 TFE_Op* func = TFE_NewOp(ctx, "VariableAddFunction", status); in TEST()
511 TFE_Op* func = TFE_NewOp(ctx, "VariableAddFunction", status); in TestDistributedFunctionCancellation()
Dc_api_cluster_test.cc95 TFE_Op* op = TFE_NewOp(ctx, "ReadVariableOp", status); in ReadVariable()
304 TFE_Op* op = TFE_NewOp(ctx, "DestroyResourceOp", status); in TestRemoteExecuteUpdateServerDefResourceAccess()
317 op = TFE_NewOp(ctx, "DestroyResourceOp", status); in TestRemoteExecuteUpdateServerDefResourceAccess()
Dcustom_device_testutil.cc117 TFE_Op* op(TFE_NewOp(context, operation_name, s)); in LoggingDeviceExecute()
Dc_api_remote_test_util.cc145 matmul = TFE_NewOp(ctx, "MatMulFunction", status); in TestRemoteExecuteSilentCopies()
Dc_api.h248 TF_CAPI_EXPORT extern TFE_Op* TFE_NewOp(TFE_Context* ctx,
Dc_api.cc617 TFE_Op* TFE_NewOp(TFE_Context* ctx, const char* op_or_function_name, in TFE_NewOp() function
694 TFE_Op* op = TFE_NewOp(ctx, op_or_function_name, status); in TFE_OpNameGetAttrType()
966 TFE_Op* func_op = TFE_NewOp(ctx, func.name().data(), status); in GetFunc()
Dc_api_experimental_test.cc289 TFE_Op* op = TFE_NewOp(ctx, "ident", status); in TEST()
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_testlib.cc38 TFE_NewOp(context, "VarHandleOp", status), TFE_DeleteOp); in Create()
63 TFE_NewOp(context, "DestroyResourceOp", status), &TFE_DeleteOp); in Destroy()
80 TFE_NewOp(context, "ReadVariableOp", status), &TFE_DeleteOp); in Read()
99 TFE_NewOp(context, op_name, status), &TFE_DeleteOp); in GeneralAssignment()
163 TFE_NewOp(context, "TPUReplicatedOutput", status), TFE_DeleteOp); in ExtractPerDeviceValues()
185 TFE_NewOp(context, "Mul", status), TFE_DeleteOp); in Multiply()
Dparallel_device_lib_test.cc57 TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp); in TEST()
75 TFE_NewOp(context.get(), "ReadVariableOp", status.get()), TFE_DeleteOp); in TEST()
113 TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp); in TEST()
185 TFE_NewOp(context.get(), "AssertAndCollective", status.get()), in TEST()
277 TFE_NewOp(context.get(), "Size", status.get()), TFE_DeleteOp); in TEST()
Dparallel_device_testlib.h122 TFE_NewOp(context, "TPUReplicatedInput", status), TFE_DeleteOp); in CreatePerDeviceValues()
Dparallel_device_lib.h262 TFE_NewOp(context, "Const", status), TFE_DeleteOp); in ScalarsFromSequence()
Dparallel_device_test.cc359 TFE_NewOp(context.get(), "TPUReplicatedOutput", status.get()), in TEST()
379 TFE_NewOp(context, "CollectiveReduce", status), TFE_DeleteOp); in CollectiveSum()
542 TFE_NewOp(context.get(), function_name, status.get()), TFE_DeleteOp); in TEST()
Dparallel_device.cc112 OpPtr op(TFE_NewOp(context, operation_name, status)); in ExecuteWithSpecialOps()
Dparallel_device_lib.cc248 op_.reset(TFE_NewOp(context, operation_name, status)); in Execute()
/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/
Dc_api_coordination_test.cc207 TFE_Op* set_op = TFE_NewOp(ctx, "TestSetConfigKeyValue", status); in TEST()
224 TFE_Op* get_op = TFE_NewOp(ctx, "TestGetConfigKeyValue", status); in TEST()
466 TFE_Op* get_op = TFE_NewOp(ctx, "TestGetConfigKeyValue", status); in TEST_P()
490 TFE_Op* set_op = TFE_NewOp(ctx, "TestSetConfigKeyValue", status); in TEST_P()
508 TFE_Op* get_op2 = TFE_NewOp(ctx, "TestGetConfigKeyValue", status); in TEST_P()
532 TFE_Op* set_fn = TFE_NewOp(ctx, "SetConfigKeyValueFn", status); in TEST_P()
553 TFE_Op* get_fn = TFE_NewOp(ctx, "GetConfigKeyValueFn", status); in TEST_P()
Dc_api_multi_client_function_test.cc262 TFE_Op* send_func = TFE_NewOp(ctx, "SendFunction", status); in TEST_P()
305 TFE_Op* recv_func = TFE_NewOp(ctx, "RecvFunction", status); in TEST_P()
/external/tensorflow/tensorflow/c/
Dc_api_experimental_test.cc166 matmul_op = TFE_NewOp(tfe_context_, "MatMul", status_); in TEST_F()
205 TFE_Op* reshape_op = TFE_NewOp(tfe_context_, "Reshape", status_); in TEST_F()
216 TFE_Op* fill_op = TFE_NewOp(tfe_context_, "Fill", status_); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Deval_util.cc87 TFE_Op* op = TFE_NewOp(context, node_def->op().c_str(), status); in EvaluateOperation()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc238 TFE_Op* op = TFE_NewOp(ctx, op_name, out_status); in EagerCast()
271 TFE_NewOp(ctx, op_name, out_status), TFE_DeleteOp); in EagerConst()
Dpywrap_tfe_src.cc520 funcs[i] = TFE_NewOp(ctx, func_name.data(), status); in SetOpAttrList()
538 TFE_Op* func_op = TFE_NewOp(ctx, func.name().data(), status); in GetFunc()
3394 TFE_Op* op = TFE_NewOp(parent_op_exec_info.ctx, "ReadVariableOp", status);
/external/tensorflow/tensorflow/java/src/main/native/
Deager_operation_builder_jni.cc73 TFE_Op* op = TFE_NewOp(context, op_or_function_name, status); in Java_org_tensorflow_EagerOperationBuilder_allocate()
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_device.cc649 TFE_NewOp(context, operation_name, status), TFE_DeleteOp); in MaybeHandleDTensorCustomOps()