Home
last modified time | relevance | path

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

/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.cc56 TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp); in TEST()
74 TFE_NewOp(context.get(), "ReadVariableOp", status.get()), TFE_DeleteOp); in TEST()
112 TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp); in TEST()
184 TFE_NewOp(context.get(), "AssertAndCollective", status.get()), in TEST()
276 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.h241 TFE_NewOp(context, "Const", status), TFE_DeleteOp); in ScalarsFromSequence()
Dparallel_device_test.cc366 TFE_NewOp(context.get(), "TPUReplicatedOutput", status.get()), in TEST()
386 TFE_NewOp(context, "CollectiveReduce", status), TFE_DeleteOp); in CollectiveSum()
551 TFE_NewOp(context.get(), function_name, status.get()), TFE_DeleteOp); in TEST()
Dparallel_device.cc110 OpPtr op(TFE_NewOp(context, operation_name, status)); in ExecuteWithSpecialOps()
Dparallel_device_lib.cc229 op_.reset(TFE_NewOp(context, operation_name, status)); in Execute()
/external/tensorflow/tensorflow/c/eager/
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_util.cc199 TFE_Op* op = TFE_NewOp(ctx, "VarHandleOp", status); in TestVariable()
218 op = TFE_NewOp(ctx, "AssignVariableOp", status); in TestVariable()
250 TFE_Op* op = TFE_NewOp(ctx, "AddV2", status); in AddOp()
265 TFE_Op* op = TFE_NewOp(ctx, "MatMul", status); in MatMulOp()
280 TFE_Op* op = TFE_NewOp(ctx, "Identity", status); in IdentityOp()
293 TFE_Op* op = TFE_NewOp(ctx, "Shape", status); in ShapeOp()
321 TFE_Op* op = TFE_NewOp(ctx, "Min", status); in MinOp()
Dc_api_test.cc85 TFE_Op* matmul = TFE_NewOp(ctx, "MatMul", status); in BM_Execute()
127 TFE_Op* identity = TFE_NewOp(ctx, "Identity", status); in BM_Execute_Identity()
1024 TFE_Op* op = TFE_NewOp(ctx, "MatMulFunction", status); in FunctionDefAndExecute()
1070 TFE_Op* op = TFE_NewOp(ctx, "AddFunction", status); in RunAddFunction()
1175 TFE_Op* matmul = TFE_NewOp(ctx, "MatMulFunction", status); in BM_ExecuteFunction()
1212 TFE_Op* op = TFE_NewOp(ctx, "ReadVariableOp", status); in TEST()
1257 TFE_Op* op = TFE_NewOp(ctx, "ReadVariableOp", status); in BM_ReadVariable()
1292 TFE_Op* op = TFE_NewOp(ctx, "AvgPool", status); in TEST()
1351 TFE_Op* op = TFE_NewOp(ctx, "AvgPool", status); in TEST()
1449 TFE_Op* minOp = TFE_NewOp(ctx, "Min", 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()
Dc_api_remote_test_util.cc145 matmul = TFE_NewOp(ctx, "MatMulFunction", status); in TestRemoteExecuteSilentCopies()
Dcustom_device_testutil.cc109 TFE_Op* op(TFE_NewOp(context, operation_name, s)); in LoggingDeviceExecute()
Dc_api.h248 TF_CAPI_EXPORT extern TFE_Op* TFE_NewOp(TFE_Context* ctx,
Dc_api.cc636 TFE_Op* TFE_NewOp(TFE_Context* ctx, const char* op_or_function_name, in TFE_NewOp() function
713 TFE_Op* op = TFE_NewOp(ctx, op_or_function_name, status); in TFE_OpNameGetAttrType()
980 TFE_Op* func_op = TFE_NewOp(ctx, func.name().data(), status); in GetFunc()
Dc_api_experimental_test.cc287 TFE_Op* op = TFE_NewOp(ctx, "ident", status); in TEST()
/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/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/python/eager/
Dpywrap_tensor.cc236 TFE_Op* op = TFE_NewOp(ctx, op_name, out_status); in EagerCast()
Dpywrap_tfe_src.cc511 funcs[i] = TFE_NewOp(ctx, func_name.data(), status); in SetOpAttrList()
529 TFE_Op* func_op = TFE_NewOp(ctx, func.name().data(), status); in GetFunc()
3315 TFE_Op* op = TFE_NewOp(parent_op_exec_info.ctx, "ReadVariableOp", status);