Home
last modified time | relevance | path

Searched defs:node_def (Results 1 – 25 of 100) sorted by relevance

1234

/external/tensorflow/tensorflow/python/framework/
Derrors_impl.py58 def __init__(self, node_def, op, message, error_code): argument
107 def node_def(self): member in OpError
219 def __init__(self, node_def, op, message): argument
238 def __init__(self, node_def, op, message, error_code=UNKNOWN): argument
258 def __init__(self, node_def, op, message): argument
273 def __init__(self, node_def, op, message): argument
291 def __init__(self, node_def, op, message): argument
308 def __init__(self, node_def, op, message): argument
326 def __init__(self, node_def, op, message): argument
341 def __init__(self, node_def, op, message): argument
[all …]
Dfunction_def_to_graph.py196 def _get_num_args(arg_def, node_def): argument
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc115 NodeDef node_def; local
402 const NodeDef& node_def, int output_port, in ConvertToTensorOrWeights()
424 NodeDef node_def = MakeConstNodeDef<float>("my_const", {1.0f, 2.0f}); in TEST_F() local
445 const NodeDef& node_def = add.operation.node()->def(); in TEST_F() local
491 NodeDef node_def = MakeNodeDef("my_op", "MyOp", {}); in TEST_F() local
547 Status GetInputs(const NodeDef& node_def, in GetInputs()
590 NodeDef node_def = MakeNodeDef("my_op", "MyOp", {"my_input"}); in TEST_F() local
614 NodeDef node_def; in TEST_F() local
670 NodeDef node_def = MakeNodeDef("my_op", "MyOp", {"my_input"}); in TEST_F() local
1240 void RunValidation(const NodeDef& node_def, in RunValidation()
[all …]
Dconvert_nodes.cc894 const NodeDef& node_def, int output_port, in ConvertToTensorOrWeights()
936 const NodeDef& node_def, in ValidateNode()
1001 Status Converter::ConvertNode(const NodeDef& node_def) { in ConvertNode()
1381 Status Converter::GetInputs(const NodeDef& node_def, in GetInputs()
1430 const auto& node_def = params.node_def; in CheckInputsWeights() local
1458 const auto& node_def = params.node_def; in AllowDataTypes() local
1576 const auto& node_def = params->node_def; in BinaryTensorOpWeight() local
1766 const auto& node_def = params->node_def; in ConvertConv2DHelper() local
1938 const auto& node_def = params->node_def; in BinaryTensorOpTensor() local
2006 const auto& node_def = params->node_def; in ConvertPlugin() local
[all …]
/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.cc48 AttrSlice::AttrSlice(const NodeDef& node_def) in AttrSlice()
88 string SummarizeNodeDef(const NodeDef& node_def) { in SummarizeNodeDef()
105 string SummarizeAttrs(const NodeDef& node_def) { in SummarizeAttrs()
119 string FormatNodeDefForError(const NodeDef& node_def) { in FormatNodeDefForError()
298 bool HasNodeAttr(const NodeDef& node_def, StringPiece attr_name) { in HasNodeAttr()
347 Status AddArgToSig(const NodeDef& node_def, const OpDef::ArgDef& arg_def, in AddArgToSig()
402 Status InputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypeForNode()
417 Status InputTypesForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypesForNode()
425 Status OutputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in OutputTypeForNode()
440 Status OutputTypesForNode(const NodeDef& node_def, const OpDef& op_def, in OutputTypesForNode()
[all …]
Dnode_def_util_test.cc41 NodeDef node_def; in ToNodeDef() local
47 NodeDef node_def; in ToNodeDef() local
79 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
144 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
164 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
190 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
220 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
279 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
379 const NodeDef node_def = ToNodeDef( in TEST() local
400 const NodeDef node_def = ToNodeDef( in TEST() local
[all …]
Dfake_input.cc60 const NodeDef* node_def, NodeDefBuilder* builder) in FakeInputImpl()
201 return [dt](const OpDef& op_def, int in_index, const NodeDef& node_def, in FakeInput()
210 return [n](const OpDef& op_def, int in_index, const NodeDef& node_def, in FakeInput()
219 return [n, dt](const OpDef& op_def, int in_index, const NodeDef& node_def, in FakeInput()
232 return [dtv](const OpDef& op_def, int in_index, const NodeDef& node_def, in FakeInput()
233 NodeDefBuilder* builder) { in FakeInput()
Dgraph_def_util.cc71 NodeDef* node_def = graph_def->mutable_node(i); in AddDefaultAttrsToGraphDef() local
85 NodeDef* node_def, const OpRegistryInterface& consumer_op_registry, in RemoveNewDefaultAttrsFromNodeDef()
143 NodeDef* node_def = graph_def->mutable_node(n); in RemoveNewDefaultAttrsFromGraphDef() local
153 NodeDef* node_def = func_def->mutable_node_def(n); in RemoveNewDefaultAttrsFromGraphDef() local
Dmemory_types_test.cc56 NodeDef node_def; in TEST() local
Dfunction.proto74 repeated NodeDef node_def = 3; field
Dgraph_to_functiondef.cc162 NodeDef* node_def = fdef->add_node_def(); in GraphToFunctionDef() local
232 NodeDef* node_def = fdef->mutable_node_def(n_index); in GraphToFunctionDef() local
Dnode_def_util.h118 NodeDef* node_def) { in AddNodeAttr()
Dop_kernel_test.cc140 NodeDef node_def; in CreateNodeDef() local
173 NodeDef node_def; in ExpectFailure() local
287 NodeDef node_def = CreateNodeDef("Test1", {DT_FLOAT, DT_INT32}); in TEST_F() local
296 NodeDef node_def = CreateNodeDef("Test1", {DT_FLOAT, DT_INT32}); in TEST_F() local
409 NodeDef node_def; in CreateNodeDef() local
932 void BM_InputRangeHelper(int iters, const NodeDef& node_def, in BM_InputRangeHelper()
961 NodeDef node_def; in BM_ConcatInputRange() local
984 NodeDef node_def; in BM_SelectInputRange() local
Dshape_inference_testutil.h37 NodeDef node_def; member
/external/tensorflow/tensorflow/core/graph/
Dgraph_constructor.cc49 inline bool IsMerge(const NodeDef& node_def) { in IsMerge()
53 inline bool IsNextIteration(const NodeDef& node_def) { in IsNextIteration()
442 const NodeDef& node_def = *node_defs_[n]; in BuildNodeIndex() local
486 const NodeDef& node_def = *node_defs[n]; in GetNextIterationNodes() local
504 const NodeDef& node_def = *node_defs_[n]; in InitFromEdges() local
558 const NodeDef& node_def) { in ValidateColocationConstraints()
575 Status GraphConstructor::MakeNode(const NodeDef& node_def, Node** node) { in MakeNode()
682 Status GraphConstructor::ModifyNodeDefForImport(NodeDef* node_def) { in ModifyNodeDefForImport()
693 void RemoveInputs(const std::vector<int>& inputs_to_remove, NodeDef* node_def, in RemoveInputs()
715 NodeDef* node_def, std::vector<bool>* input_already_exists) { in RemapNodeDefInputs()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dcreate_xla_launch_op.cc69 const NodeDef& node_def) { in CompilationRequested()
115 const NodeDef& node_def, in GetBodyAndConstantsAndResources()
153 Status CreateXlaLaunchOp(FunctionLibraryRuntime* flr, const NodeDef& node_def, in CreateXlaLaunchOp()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dsharding_util.cc30 const NodeDef& node_def) { in GetShardingFromNodeDef()
81 const NodeDef& node_def, int num_cores_per_replica) { in ParseShardingFromDevice()
Dfunctionalize_control_flow_util.cc32 xla::StatusOr<Node*> AddNodeDefToGraph(const NodeDef& node_def, Graph* graph) { in AddNodeDefToGraph()
/external/tensorflow/tensorflow/core/util/
Dmirror_pad_mode.cc24 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name, in GetNodeAttr()
Dpadding.cc24 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name, in GetNodeAttr()
/external/tensorflow/tensorflow/python/util/
Dkernel_registry.cc29 tensorflow::NodeDef node_def; in TryFindKernelClass() local
/external/tensorflow/tensorflow/contrib/compiler/
Djit.py81 def xla_compile(node_def): argument
/external/tensorflow/tensorflow/contrib/hvx/hvx_ops_support_checker/
Dhvx_ops_support_checker_main.cc57 static void SummarizeNode(const NodeDef& node_def, in SummarizeNode()
78 static void DumpRemoteFusedGraph(const NodeDef& node_def) { in DumpRemoteFusedGraph()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Dvectorizer_registry_test.cc41 NodeDef node_def; in TEST() local
/external/tensorflow/tensorflow/core/kernels/
Dops_testutil.h90 void set_node_def(const NodeDef& node_def) { node_def_.CopyFrom(node_def); } in set_node_def()
93 NodeDef* node_def() { return &node_def_; } in node_def() function

1234