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.cc73 Status s = op_registry.LookUpOpDef(node_def->op(), &op_def); in AddDefaultAttrsToGraphDef()
91 producer_op_registry.LookUpOpDef(node_def->op(), &producer_op_def)); in RemoveNewDefaultAttrsFromNodeDef()
93 consumer_op_registry.LookUpOpDef(node_def->op(), &consumer_op_def)); in RemoveNewDefaultAttrsFromNodeDef()
221 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(op_name, &op_def)); in StrippedOpListForGraph()
Dfunction_test.cc53 return OpRegistry::Global()->LookUpOpDef(op, sig); in GetOpSig()
995 EXPECT_TRUE(!lib_def.LookUpOpDef("XTimes16", &op_def).ok()); in TEST()
997 TF_EXPECT_OK(lib_def.LookUpOpDef("XTimesTwo", &op_def)); in TEST()
1020 TF_EXPECT_OK(lib_def.LookUpOpDef("XTimesTwo", &first)); in TEST()
1027 TF_EXPECT_OK(lib_def.LookUpOpDef("WXPlusB", &second)); in TEST()
1249 TF_EXPECT_OK(lib_def1.LookUpOpDef("XTimesTwo", &f1)); in TEST()
1250 TF_EXPECT_OK(lib_def2.LookUpOpDef("XTimesTwo", &f2)); in TEST()
1254 TF_EXPECT_OK(lib_def1.LookUpOpDef("WXPlusB", &f3)); in TEST()
1255 TF_EXPECT_OK(lib_def2.LookUpOpDef("WXPlusB", &f4)); in TEST()
Dmemory_types.cc81 TF_RETURN_IF_ERROR(op_registry->LookUpOpDef(ndef.op(), &op_def)); in MemoryTypesForNode()
Dop.h52 Status LookUpOpDef(const string& op_type_name, const OpDef** op_def) const;
Dop.cc37 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/jit/
Dextract_outside_compilation_pass_test.cc307 return fld.LookUpOpDef(op, sig); in TEST_F()
354 return fld.LookUpOpDef(op, sig); in TEST_F()
439 return fld.LookUpOpDef(op, sig); in TEST_F()
483 return fld.LookUpOpDef(op, sig); in TEST_F()
500 return fld.LookUpOpDef(op, sig); in TEST_F()
611 return fld.LookUpOpDef(op, sig); in TEST_F()
661 return fld.LookUpOpDef(op, sig); in TEST_F()
770 return fld.LookUpOpDef(op, sig); in TEST_F()
910 return fld.LookUpOpDef(op, sig); in TEST_F()
925 return fld.LookUpOpDef(op, sig); in TEST_F()
[all …]
Dextract_outside_compilation_pass.cc500 return fld->LookUpOpDef(op, sig); in ConstructHostGraph()
620 return fld->LookUpOpDef(op, sig); in ExpandHostGraphIntoMainGraph()
697 return fld->LookUpOpDef(op, sig); in RewriteShapeInferenceGraph()
837 return fld->LookUpOpDef(op, sig); in ReplaceKeyPlaceholderWithArgNode()
968 return fld->LookUpOpDef(op, sig); in AddSendLoopPredToLoopCond()
1039 return fld->LookUpOpDef(op, sig); in RewriteHostWhileLoopCond()
1119 return fld->LookUpOpDef(op, sig); in RewriteHostWhileLoopBody()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.cc584 Status status = OpRegistry::Global()->LookUpOpDef(node.op(), &op_def); in MaybeHasRefInput()
607 Status status = op_registry->LookUpOpDef(op_name, &op_def); in IsStateful()
628 Status status = op_registry->LookUpOpDef(op_name, &op_def); in IsFreeOfSideEffect()
689 Status status = OpRegistry::Global()->LookUpOpDef(node.op(), &op_def); \
812 return OpRegistry::Global()->LookUpOpDef(node.op(), &op_def).ok(); in HasOpDef()
Dgraph_view_test.cc48 TF_EXPECT_OK(OpRegistry::Global()->LookUpOpDef(a_node_def.op(), &a_op_def)); in TEST_F()
49 TF_EXPECT_OK(OpRegistry::Global()->LookUpOpDef(b_node_def.op(), &b_op_def)); in TEST_F()
80 TF_EXPECT_OK(OpRegistry::Global()->LookUpOpDef(b_node_def.op(), &b_op_def)); in TEST_F()
113 TF_EXPECT_OK(OpRegistry::Global()->LookUpOpDef(c_node_def.op(), &c_op_def)); in TEST_F()
Dutils.cc234 auto status = OpRegistry::Global()->LookUpOpDef(node.op(), &op_def); in NumOutputs()
248 auto status = fdef.LookUpOpDef(node.op(), &op_def); in NumOutputs()
/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()
211 Status lookup_status = op_registry->LookUpOpDef(op_name, &op_def); in RegisterCompilationKernels()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dpin_to_host_optimizer.cc141 Status s = OpRegistry::Global()->LookUpOpDef(node.op(), &op); in IsNodeOutputPortHostFriendly()
186 Status s = OpRegistry::Global()->LookUpOpDef(node.op(), &op); in IsNodeInputPortHostFriendly()
Dmemory_optimizer.cc724 TF_RETURN_IF_ERROR(OpRegistry::Global()->LookUpOpDef(node->op(), &op_def)); in BuildSwapPair()
864 if (!OpRegistry::Global()->LookUpOpDef(node.op(), &op_def).ok()) { in IsSwappable()
928 if (!OpRegistry::Global()->LookUpOpDef(node.op(), &op_def).ok()) { in IsSwappable()
/external/tensorflow/tensorflow/core/graph/
Dvalidate.cc38 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(node_def.op(), &op_def)); in ValidateGraphDef()
Dgraph_constructor.cc684 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def->op(), &op_def)); in ModifyNodeDefForImport()
899 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def.op(), &op_def)); in IsNodeFullyMapped()
1171 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def->op(), &op_def)); in PopulateMissingUnusedInputMapKeys()
Dgraph.cc408 status->Update(ops_.LookUpOpDef(node_def.op(), &op_def)); in AddNode()
435 TF_CHECK_OK(ops_.LookUpOpDef(node->type_string(), &op_def)); in CopyNode()
/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.cc187 Status s = OpRegistry::Global()->LookUpOpDef(node.op(), &op_def); in IsNodeStateful()
Dhoist_random_uniform.cc191 TF_CHECK_OK(library.LookUpOpDef(node.op(), &op_def)); in CanHoistRandomUniform()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_if_op.cc247 return flib.LookUpOpDef(op, sig); in InlineCallInGraph()
Dlower_while_op.cc384 return lib.LookUpOpDef(op, sig); in InlineCallInGraph()
/external/tensorflow/tensorflow/python/grappler/
Ditem.i265 tensorflow::Status s = registry->LookUpOpDef(node.op(), &op_def); in TF_GetColocationGroups()

123