Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/python/framework/
Derrors_impl.py39 def __init__(self, node_def, op, message, error_code): argument
82 def node_def(self): member in OpError
194 def __init__(self, node_def, op, message): argument
213 def __init__(self, node_def, op, message, error_code=UNKNOWN): argument
233 def __init__(self, node_def, op, message): argument
248 def __init__(self, node_def, op, message): argument
266 def __init__(self, node_def, op, message): argument
283 def __init__(self, node_def, op, message): argument
301 def __init__(self, node_def, op, message): argument
316 def __init__(self, node_def, op, message): argument
[all …]
Dimporter.py43 def _GetNodeAttr(node_def, attr_name): argument
50 def _ArgToTypesNoRef(node_def, arg_def): argument
68 def _SingleArgToTypes(node_def, arg_def): argument
75 def _ArgsToTypes(node_def, arg_list): argument
82 def _InputTypes(node_def, op_dict): argument
87 def _OutputTypes(node_def, op_dict): argument
387 def _SetDefaultAttrValues(node_def, op_def): argument
/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.cc47 AttrSlice::AttrSlice(const NodeDef& node_def) in AttrSlice()
87 string SummarizeNodeDef(const NodeDef& node_def) { in SummarizeNodeDef()
246 bool HasNodeAttr(const NodeDef& node_def, StringPiece attr_name) { in HasNodeAttr()
295 Status AddArgToSig(const NodeDef& node_def, const OpDef::ArgDef& arg_def, in AddArgToSig()
350 Status InputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypeForNode()
365 Status OutputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in OutputTypeForNode()
380 Status InOutTypesForNode(const NodeDef& node_def, const OpDef& op_def, in InOutTypesForNode()
391 Status ValidateNodeDef(const NodeDef& node_def, const OpDef& op_def) { in ValidateNodeDef()
483 Status ComputeArgRange(const NodeDef& node_def, const OpDef::ArgDef& arg_def, in ComputeArgRange()
503 Status NameRangesHelper(const NodeDef& node_def, in NameRangesHelper()
[all …]
Dnode_def_util_test.cc38 NodeDef node_def; in ToNodeDef() local
44 NodeDef node_def; in ToNodeDef() local
76 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
141 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
161 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
187 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
216 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
274 const NodeDef node_def = ToNodeDef(R"proto( in TEST() local
375 const NodeDef node_def = ToNodeDef( in TEST() local
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.cc64 NodeDef* node_def = graph_def->mutable_node(i); in AddDefaultAttrsToGraphDef() local
74 NodeDef* node_def, const OpRegistryInterface& consumer_op_registry, in RemoveNewDefaultAttrsFromNodeDef()
132 NodeDef* node_def = graph_def->mutable_node(n); in RemoveNewDefaultAttrsFromGraphDef() local
142 NodeDef* node_def = func_def->mutable_node_def(n); in RemoveNewDefaultAttrsFromGraphDef() local
Dfunction.proto73 repeated NodeDef node_def = 3; field
Dmemory_types_test.cc56 NodeDef node_def; in TEST() local
Dnode_def_util.h105 NodeDef* node_def) { in AddNodeAttr()
Dop_kernel_test.cc119 NodeDef node_def; in CreateNodeDef() local
152 NodeDef node_def; in ExpectFailure() local
254 NodeDef node_def = CreateNodeDef("Test1", {DT_FLOAT, DT_INT32}); in TEST_F() local
263 NodeDef node_def = CreateNodeDef("Test1", {DT_FLOAT, DT_INT32}); in TEST_F() local
376 NodeDef node_def; in CreateNodeDef() local
898 void BM_InputRangeHelper(int iters, const NodeDef& node_def, in BM_InputRangeHelper()
927 NodeDef node_def; in BM_ConcatInputRange() local
950 NodeDef node_def; in BM_SelectInputRange() local
Dshape_inference_testutil.h36 NodeDef node_def; member
/external/tensorflow/tensorflow/core/graph/
Dgraph_constructor.cc46 inline bool IsMerge(const NodeDef& node_def) { in IsMerge()
50 inline bool IsNextIteration(const NodeDef& node_def) { in IsNextIteration()
414 const NodeDef& node_def = *node_defs_[n]; in BuildNodeIndex() local
458 const NodeDef& node_def = *node_defs[n]; in GetNextIterationNodes() local
476 const NodeDef& node_def = *node_defs_[n]; in InitFromEdges() local
530 const NodeDef& node_def) { in ValidateColocationConstraints()
547 Status GraphConstructor::MakeNode(const NodeDef& node_def, Node** node) { in MakeNode()
645 Status GraphConstructor::ModifyNodeDefForImport(NodeDef* node_def) { in ModifyNodeDefForImport()
656 void RemoveInputs(const std::vector<int>& inputs_to_remove, NodeDef* node_def, in RemoveInputs()
681 NodeDef* node_def, std::vector<bool>* input_already_exists) { in RemapNodeDefInputs()
[all …]
Dvalidate.cc33 for (const NodeDef& node_def : graph_def.node()) { in ValidateGraphDef() local
Dgraph.cc38 NodeProperties(const OpDef* op_def, const NodeDef& node_def, in NodeProperties()
46 NodeDef node_def; member in tensorflow::NodeProperties
323 Node* Graph::AddNode(const NodeDef& node_def, Status* status) { in AddNode()
543 NodeDef* node_def = graph_def->add_node(); in ToGraphDefSubRange() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dsharding_util.cc30 GetShardingFromNodeDef(const NodeDef& node_def) { in GetShardingFromNodeDef()
82 ParseShardingFromDevice(const NodeDef& node_def, int num_cores_per_replica) { in ParseShardingFromDevice()
/external/tensorflow/tensorflow/contrib/tensorrt/convert/
Dconvert_nodes.cc343 const tensorflow::NodeDef& node_def) { in get_inputs()
371 tensorflow::Status convert_node(const tensorflow::NodeDef& node_def) { in convert_node()
601 Converter& ctx, const tensorflow::NodeDef& node_def, in ConstantFoldUnary()
637 Converter& ctx, const tensorflow::NodeDef& node_def, in ConstantFoldBinary()
707 Converter& ctx, const tensorflow::NodeDef& node_def, in BinaryTensorOpWeight()
789 Converter& ctx, const tensorflow::NodeDef& node_def, in BinaryTensorOpTensor()
828 Converter& ctx, const tensorflow::NodeDef& node_def, in ConvertPlaceholder()
851 const tensorflow::NodeDef& node_def, in ConvertConv2D()
939 const tensorflow::NodeDef& node_def, in ConvertPool()
1021 Converter& ctx, const tensorflow::NodeDef& node_def, in ConvertActivation()
[all …]
/external/tensorflow/tensorflow/core/util/
Dpadding.cc24 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name, in GetNodeAttr()
Dmirror_pad_mode.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/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
Dqueue_base.cc92 Status QueueBase::MatchesNodeDefOp(const NodeDef& node_def, in MatchesNodeDefOp()
102 Status QueueBase::MatchesNodeDefCapacity(const NodeDef& node_def, in MatchesNodeDefCapacity()
/external/tensorflow/tensorflow/compiler/jit/
Dgraph_to_functiondef.cc161 NodeDef* node_def = fdef->add_node_def(); in GraphToFunctionDef() local
230 NodeDef* node_def = fdef->mutable_node_def(n_index); in GraphToFunctionDef() local
/external/tensorflow/tensorflow/python/tools/
Doptimize_for_inference_lib.py173 def values_from_const(node_def): argument

123