/external/tensorflow/tensorflow/core/graph/ |
D | validate.cc | 30 const OpRegistryInterface& op_registry) { in ValidateGraphDef() argument 36 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(node_def.op(), &op_def)); in ValidateGraphDef() 45 const GraphDef& graph_def, const OpRegistryInterface& op_registry) { in ValidateGraphDefAgainstOpRegistry() argument 47 TF_RETURN_IF_ERROR(AddDefaultAttrsToGraphDef(©, op_registry, 0)); in ValidateGraphDefAgainstOpRegistry() 48 return ValidateGraphDef(copy, op_registry); in ValidateGraphDefAgainstOpRegistry() 57 void GetOpListForValidation(OpList* op_list, const OpRegistry& op_registry) { in GetOpListForValidation() argument 58 op_registry.Export(false, op_list); in GetOpListForValidation()
|
D | graph_def_builder.h | 120 const OpRegistryInterface* op_registry() const { in op_registry() function 121 return graph_->op_registry(); in op_registry() 146 const OpRegistryInterface* op_registry = OpRegistry::Global()) 147 : graph_(op_registry), opts_(&graph_, &status_) {} in graph_() argument 154 const OpRegistryInterface* op_registry = OpRegistry::Global()) 155 : graph_(op_registry), opts_(&graph_, nullptr) {} in graph_() argument
|
D | validate.h | 34 const OpRegistryInterface& op_registry); 40 const GraphDef& graph_def, const OpRegistryInterface& op_registry); 51 OpList* op_list, const OpRegistry& op_registry = *OpRegistry::Global());
|
D | graph_constructor_test.cc | 1167 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() 1207 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() 1270 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() 1335 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() 1367 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() 1388 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() 1410 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() 1435 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() 1507 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() 1544 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() [all …]
|
D | graph_def_builder.cc | 104 opts.op_registry()); in SourceOp() 112 opts.op_registry()); in UnaryOp() 121 opts.op_registry()); in BinaryOp()
|
D | testlib.cc | 152 TF_CHECK_OK(NodeBuilder(g->NewName("n"), reduce, g->op_registry()) in Reduce() 197 TF_CHECK_OK(NodeBuilder(g->NewName("n"), op, g->op_registry()) in RandomNumberGenerator() 239 TF_CHECK_OK(NodeBuilder(g->NewName("n"), func, g->op_registry()) in Unary() 247 TF_CHECK_OK(NodeBuilder(g->NewName("n"), func, g->op_registry()) in Binary() 256 auto b = NodeBuilder(g->NewName("n"), func, g->op_registry()); in Multi() 278 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Roll", g->op_registry()) in Roll()
|
D | node_builder.cc | 39 const OpRegistryInterface* op_registry) in NodeBuilder() argument 40 : def_builder_(name, op_name, op_registry) {} in NodeBuilder()
|
D | graph_constructor.cc | 647 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def->op(), &op_def)); in ModifyNodeDefForImport() 853 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def.op(), &op_def)); in IsNodeFullyMapped() 1135 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def->op(), &op_def)); in PopulateMissingUnusedInputMapKeys() 1171 ShapeRefiner refiner(gdef.versions().producer(), g->op_registry()); in ConvertGraphDefToGraph() 1180 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, g->op_registry()); in ConvertNodeDefsToGraph() 1224 ShapeRefiner default_refiner(gdef.versions().producer(), g->op_registry()); in ImportGraphDef()
|
D | node_builder.h | 79 const OpRegistryInterface* op_registry = OpRegistry::Global());
|
/external/tensorflow/tensorflow/core/framework/ |
D | graph_def_util.cc | 54 const OpRegistryInterface& op_registry, in AddDefaultAttrsToGraphDef() argument 66 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(node_def->op(), &op_def)); in AddDefaultAttrsToGraphDef() 201 const OpRegistryInterface& op_registry, in StrippedOpListForGraph() argument 210 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(op_name, &op_def)); in StrippedOpListForGraph()
|
D | graph_def_util.h | 56 const OpRegistryInterface& op_registry, 110 const OpRegistryInterface& op_registry,
|
D | memory_types.cc | 70 Status MemoryTypesForNode(const OpRegistryInterface* op_registry, in MemoryTypesForNode() argument 76 TF_RETURN_IF_ERROR(op_registry->LookUpOpDef(ndef.op(), &op_def)); in MemoryTypesForNode()
|
D | memory_types.h | 31 Status MemoryTypesForNode(const OpRegistryInterface* op_registry,
|
D | dataset.cc | 105 opts->op_registry()); in AddDataset() 191 Status s = b_->opts().op_registry()->LookUpOpDef(op_type_name, &op_def); in HasAttr()
|
D | node_def_builder.cc | 40 const OpRegistryInterface* op_registry) { in NodeDefBuilder() argument 42 const Status status = op_registry->LookUpOpDef(op_name.ToString(), &op_def_); in NodeDefBuilder()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | tf2xla_util.cc | 98 const tf2xla::Config& config, const OpRegistryInterface* op_registry, in AddPlaceholdersForFeeds() argument 146 AddDefaultAttrsToGraphDef(&gd, *op_registry, 0 /*node_offset*/)); in AddPlaceholdersForFeeds() 149 Graph g(op_registry); in AddPlaceholdersForFeeds()
|
D | xla_op_registry.cc | 159 OpRegistryInterface* op_registry = OpRegistry::Global(); in RegisterCompilationKernels() local 164 Status lookup_status = op_registry->LookUpOpDef(op_name, &op_def); in RegisterCompilationKernels() 169 dynamic_cast<OpRegistry*>(op_registry)->DebugString(true)); in RegisterCompilationKernels()
|
D | tf2xla_util.h | 38 const tf2xla::Config& config, const OpRegistryInterface* op_registry,
|
D | tf2xla.cc | 328 TF_RETURN_IF_ERROR(AddPlaceholdersForFeeds(config, g->op_registry(), in InitGraph() 338 &second_copy_def, *g->op_registry(), /*node_offset=*/0)); in InitGraph()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | graph_rewriter.cc | 32 OpRegistryInterface* op_registry = OpRegistry::Global(); in GraphRewriter() local 38 Status s = op_registry->LookUp(node.op(), &op_reg_data); in GraphRewriter()
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | mark_for_compilation_pass_test.cc | 55 FunctionLibraryDefinition flib_def((*graph)->op_registry(), flib); in MarkForCompilation() 177 builder.opts().op_registry()); in TEST() 312 builder.opts().op_registry()); in TEST() 325 builder.opts().op_registry()); in TEST()
|
/external/tensorflow/tensorflow/contrib/coder/kernels/ |
D | range_coder_ops_test.cc | 331 TF_ASSERT_OK(NodeBuilder("range_decode", "RangeDecode", g.op_registry()) in TEST_F() 338 ShapeRefiner refiner{g.versions().producer(), g.op_registry()}; in TEST_F() 432 TF_CHECK_OK(NodeBuilder("range_encode", "RangeEncode", g->op_registry()) in CreateRangeEncodeFullBroadcastGraph() 481 TF_CHECK_OK(NodeBuilder("range_decode", "RangeDecode", g->op_registry()) in CreateRangeDecodeFullBroadcastGraph()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | memory_types.cc | 66 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type, in ProcessMemoryTypes() 208 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type, n->def(), in MemoryTypeForOutput()
|
D | function_testlib.cc | 62 NodeDefBuilder builder(op_name, fn_name, scope->graph()->op_registry()); in Call()
|
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | utils.cc | 86 OpRegistry* op_registry = OpRegistry::Global(); in ExtractExtraProperties() local 88 auto s = op_registry->LookUpOpDef(node.op(), &op_def); in ExtractExtraProperties()
|