Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/framework/
Dgraph_def_util.cc74 Status s = op_registry.LookUpOpDef(node_def->op(), &op_def); in AddDefaultAttrsToGraphDef()
92 producer_op_registry.LookUpOpDef(node_def->op(), &producer_op_def)); in RemoveNewDefaultAttrsFromNodeDef()
94 consumer_op_registry.LookUpOpDef(node_def->op(), &consumer_op_def)); in RemoveNewDefaultAttrsFromNodeDef()
258 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(op_name, &op_def)); in StrippedOpListForGraph()
Dnode_properties.cc28 TF_RETURN_IF_ERROR(op_registry->LookUpOpDef(node_def.op(), &op_def)); in CreateFromNodeDef()
Dfunction_test.cc56 return OpRegistry::Global()->LookUpOpDef(op, sig); in GetOpSig()
996 EXPECT_FALSE(lib_def.LookUpOpDef("XTimes16", &op_def).ok()); in TEST()
998 TF_EXPECT_OK(lib_def.LookUpOpDef("XTimesTwo", &op_def)); in TEST()
1016 TF_EXPECT_OK(lib_def.LookUpOpDef("XTimesTwo", &op_def)); in TEST()
1260 TF_EXPECT_OK(lib_def1.LookUpOpDef(name, &f1)); in TEST()
1261 TF_EXPECT_OK(lib_def2.LookUpOpDef(name, &f2)); in TEST()
Dmemory_types.cc88 TF_RETURN_IF_ERROR(op_registry->LookUpOpDef(ndef.op(), &op_def)); in MemoryTypesForNode()
Dop.h52 Status LookUpOpDef(const std::string& op_type_name,
Dop.cc43 Status OpRegistryInterface::LookUpOpDef(const string& op_type_name, in LookUpOpDef() function in tensorflow::OpRegistryInterface
Dnode_def_builder.cc44 const Status status = op_registry->LookUpOpDef(string(op_name), &op_def_); in NodeDefBuilder()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dupgrade_graph.cc85 TF_RETURN_IF_ERROR(flib_def->LookUpOpDef(node_def.op(), &op_def)); in GenerateResourceSharedNameIfEmpty()
100 TF_RETURN_IF_ERROR(flib_def->LookUpOpDef(node_def.op(), &op_def)); in GenerateResourceSharedNameIfEmpty()
/external/tensorflow/tensorflow/python/framework/
Dop_def_registry.cc27 auto status = tensorflow::OpRegistry::Global()->LookUpOpDef(name, &op_def); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtest_util.cc30 return library.LookUpOpDef(op, sig); in InstantiateFunctionForTest()
Dresource_operation_table_test.cc47 TF_ASSERT_OK(OpRegistry::Global()->LookUpOpDef(xla_op_name, &op_def)); in TEST()
Dxla_op_registry.cc44 TF_RETURN_IF_ERROR(OpRegistry::Global()->LookUpOpDef("XlaLaunch", &op_def)); in LaunchOpHasKernelForDevice()
222 Status lookup_status = op_registry->LookUpOpDef(op_name, &op_def); in RegisterCompilationKernels()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.cc673 Status status = OpRegistry::Global()->LookUpOpDef(node.op(), &op_def); in HasRefInput()
696 Status status = op_registry->LookUpOpDef(op_name, &op_def); in IsStateful()
717 Status status = op_registry->LookUpOpDef(op_name, &op_def); in IsFreeOfSideEffect()
780 Status status = OpRegistry::Global()->LookUpOpDef(node.op(), &op_def); \
903 return OpRegistry::Global()->LookUpOpDef(node.op(), &op_def).ok(); in HasOpDef()
Dgraph_view_test.cc49 TF_EXPECT_OK(OpRegistry::Global()->LookUpOpDef(a_node_def.op(), &a_op_def)); in TEST_F()
50 TF_EXPECT_OK(OpRegistry::Global()->LookUpOpDef(b_node_def.op(), &b_op_def)); in TEST_F()
81 TF_EXPECT_OK(OpRegistry::Global()->LookUpOpDef(b_node_def.op(), &b_op_def)); in TEST_F()
114 TF_EXPECT_OK(OpRegistry::Global()->LookUpOpDef(c_node_def.op(), &c_op_def)); in TEST_F()
Dutils.cc169 auto status = OpRegistry::Global()->LookUpOpDef(node.op(), &op_def); in NumOutputs()
183 auto status = fdef.LookUpOpDef(node.op(), &op_def); in NumOutputs()
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_def_utils.cc80 return lib_def->LookUpOpDef(op, sig); in FunctionDefToBodyHelper()
Dgraph_constructor.cc837 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def->op(), &op_def)); in ModifyNodeDefForImport()
1051 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def.op(), &op_def)); in IsNodeFullyMapped()
1237 g_->op_registry()->LookUpOpDef(node_def.op(), &op_def)); in Convert()
1407 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def.op(), &op_def)); in PopulateMissingUnusedInputMapKeys()
Dfunction.cc273 const Status s = lib_def_->LookUpOpDef(function_name, &op_def); in IsStateful()
481 return base_lib_def_->LookUpOpDef(op, sig); in FunctionLibraryRuntimeImpl()
680 return lib_def->LookUpOpDef(op, sig); in FunctionDefToBody()
1306 const Status s = base_lib_def_->LookUpOpDef(func, &op_def); in IsStateful()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dpin_to_host_optimizer.cc145 Status s = OpRegistry::Global()->LookUpOpDef(node.op(), &op); in IsNodeOutputPortHostFriendly()
190 Status s = OpRegistry::Global()->LookUpOpDef(node.op(), &op); in IsNodeInputPortHostFriendly()
/external/tensorflow/tensorflow/core/graph/
Dvalidate.cc38 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(node_def.op(), &op_def)); in ValidateGraphDef()
/external/tensorflow/tensorflow/core/grappler/graph_analyzer/
Dgen_node.cc52 Status st = OpRegistry::Global()->LookUpOpDef(opcode(), &op_); in ParseInputs()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfunction_utils.cc196 Status s = OpRegistry::Global()->LookUpOpDef(node.op(), &op_def); in IsNodeStateful()
Dhoist_random_uniform.cc201 TF_CHECK_OK(library.LookUpOpDef(node.op(), &op_def)); in CanHoistRandomUniform()
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dtfr_decompose_ctx.cc124 TF_RETURN_IF_ERROR(OpRegistry::Global()->LookUpOpDef(node_def.op(), &op_def)); in ExpandNode()
/external/tensorflow/tensorflow/python/grappler/
Ditem_wrapper.cc220 if (!registry->LookUpOpDef(node.op(), &op_def).ok()) { in PYBIND11_MODULE()

123