Home
last modified time | relevance | path

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

1234567

/external/tensorflow/tensorflow/python/framework/
Derrors_impl.py53 def __init__(self, node_def, op, message, error_code, *args): argument
110 def node_def(self): member in OpError
230 def __init__(self, node_def, op, message, *args): argument
252 def __init__(self, node_def, op, message, *args): argument
273 def __init__(self, node_def, op, message, *args): argument
288 def __init__(self, node_def, op, message, *args): argument
306 def __init__(self, node_def, op, message, *args): argument
323 def __init__(self, node_def, op, message, *args): argument
341 def __init__(self, node_def, op, message, *args): argument
356 def __init__(self, node_def, op, message, *args): argument
[all …]
Dnode_file_writer_test.py110 def _get_input_shapes(self, node_def): argument
117 def _get_input_dtypes(self, node_def): argument
123 def _get_input_tensor(self, node_def, input_index): argument
Ddevice.py119 def __call__(self, node_def): argument
128 def shortcut_string_merge(self, node_def): argument
/external/tensorflow/tensorflow/core/framework/
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"pb( in TEST() local
145 const NodeDef node_def = ToNodeDef(R"pb( in TEST() local
170 const NodeDef node_def = ToNodeDef(R"pb( in TEST() local
201 const NodeDef node_def = ToNodeDef(R"pb( in TEST() local
261 const NodeDef node_def = ToNodeDef(R"pb( in TEST() local
338 const NodeDef node_def = ToNodeDef(R"pb( in TEST() local
555 const NodeDef node_def = ToNodeDef(std::move( in TEST() local
576 const NodeDef node_def = ToNodeDef(std::move( in TEST() local
[all …]
Dnode_def_util.cc61 AttrSlice::AttrSlice(const NodeDef& node_def) in AttrSlice()
111 string SummarizeNodeDef(const NodeDef& node_def, int max_inputs_in_summary) { in SummarizeNodeDef()
132 string SummarizeAttrs(const NodeDef& node_def) { in SummarizeAttrs()
147 string FormatNodeDefForError(const NodeDef& node_def) { in FormatNodeDefForError()
344 bool HasNodeAttr(const NodeDef& node_def, StringPiece attr_name) { in HasNodeAttr()
530 Status InputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypeForNode()
546 Status InputTypesForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypesForNode()
554 Status OutputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in OutputTypeForNode()
570 Status OutputTypesForNode(const NodeDef& node_def, const OpDef& op_def, in OutputTypesForNode()
586 Status InOutTypesForNode(const NodeDef& node_def, const OpDef& op_def, in InOutTypesForNode()
[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()
Dnode_properties_test.cc62 const NodeDef& node_def, in ValidateNodeProperties()
83 NodeDef node_def; in TEST() local
105 NodeDef node_def; in TEST() local
Dgraph_def_util.cc72 NodeDef* node_def = graph_def->mutable_node(i); in AddDefaultAttrsToGraphDef() local
86 NodeDef* node_def, const OpRegistryInterface& consumer_op_registry, in RemoveNewDefaultAttrsFromNodeDef()
144 NodeDef* node_def = graph_def->mutable_node(n); in RemoveNewDefaultAttrsFromGraphDef() local
154 NodeDef* node_def = func_def->mutable_node_def(n); in RemoveNewDefaultAttrsFromGraphDef() local
Dnode_properties.cc25 NodeDef node_def, const OpRegistryInterface* op_registry, in CreateFromNodeDef()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
Dvariable_ops.cc163 const auto& node_def = params_->node_def; in ValidateImpl() local
198 const auto& node_def = params_->node_def; in Validate() local
226 const auto& node_def = params_->node_def; in Convert() local
267 const auto& node_def = params_->node_def; in ValidateImpl() local
296 const auto& node_def = params_->node_def; in Validate() local
329 const auto& node_def = params_->node_def; in Convert() local
Dunary_ops.cc109 const auto& node_def = params.node_def; in ConvertImpl() local
201 const auto& node_def = params_->node_def; in Validate() local
211 const auto& node_def = params_->node_def; in Convert() local
Dfill_ops.cc64 const auto& node_def = params.node_def; in Validate() local
133 const auto& node_def = params.node_def; in Validate() local
296 std::string convert_range_expected_msg(const NodeDef& node_def) { in convert_range_expected_msg()
Ddata_format_vec_permute.cc58 const auto& node_def = params_->node_def; in Validate() local
115 const auto& node_def = params_->node_def; in Convert() local
Dbinary_ops.cc67 const auto& node_def = params.node_def; local
115 const auto& node_def = params.node_def; local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc548 Status GetInputs(const NodeDef& node_def, in GetInputs()
587 NodeDef node_def = MakeNodeDef("my_op", "MyOp", {"my_input"}); in TEST_F() local
618 NodeDef node_def; in TEST_F() local
676 NodeDef node_def = MakeNodeDef("my_op", "MyOp", {"my_input"}); in TEST_F() local
1484 void RunValidationAndConversion(const NodeDef& node_def, in RunValidationAndConversion()
1516 const NodeDef& node_def, const Status& status, in RunValidationAndConversion()
1611 void CreateContext(const NodeDef& node_def, OpKernel** kernel, in CreateContext()
1924 const string& name, const NodeDef node_def, in TestOpConverterMultiOut()
1940 void TestOpConverter(const string& name, const NodeDef node_def, in TestOpConverter()
1989 const NodeDef& node_def = op_map[op_name].first(tf_type); in RunTests() local
[all …]
Dconvert_nodes.cc85 #define TFTRT_CHECK_INPUT_SIZE(size, exp_size, node_def) \ argument
839 const NodeDef& node_def, const std::vector<TRT_TensorOrWeights>& inputs, in OpConverterParams()
854 Converter* converter, const NodeDef& node_def, in OpConverterParams()
883 const NodeDef& node_def, int output_port, in ConvertToTensorOrWeights()
1067 Status Converter::ConvertNode(const NodeDef& node_def) { in ConvertNode()
1415 const NodeDef& node_def, in TransposeTensor()
1500 void Converter::SetLayerName(nvinfer1::ILayer* layer, const NodeDef& node_def, in SetLayerName()
1541 ITensorProxyPtr* tensor, const NodeDef& node_def, in PrepareTensorForShape()
1622 Status Converter::GetInputs(const NodeDef& node_def, in GetInputs()
1669 const auto& node_def = params.node_def; in CheckInputsWeights() local
[all …]
Dop_converter.h57 const NodeDef& node_def; member
114 const auto& node_def = params_->node_def; in ValidateNodeDefDataType() local
134 const NodeDef& node_def = params_->node_def; in ValidateInputs() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmake_deterministic_test.cc304 NodeDef node_def = output.node(index); in TEST_P() local
333 NodeDef node_def = output.node(index); in TEST_P() local
367 NodeDef node_def = output.node(index); in TEST_P() local
400 NodeDef node_def = output.node(index); in TEST_P() local
443 NodeDef node_def = output.node(index); in TEST_P() local
510 NodeDef node_def = output.node(index); in TEST_P() local
558 NodeDef node_def = output.node(index); in TEST_P() local
600 NodeDef node_def = output.node(index); in TEST_P() local
643 NodeDef node_def = output.node(index); in TEST() local
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_constructor.cc61 inline bool IsMerge(const NodeDef& node_def) { in IsMerge()
66 inline bool IsNextIteration(const NodeDef& node_def) { in IsNextIteration()
645 const NodeDef& node_def = get_node_def(n); in BuildNodeIndex() local
692 const NodeDef& node_def = get_node_def(n); in InitFromEdges() local
700 const NodeDef& node_def = get_node_def(n); in InitFromEdges() local
754 const NodeDef& node_def) { in ValidateColocationConstraints()
771 Status GraphConstructor::MakeNode(NodeDef&& node_def, Node** node) { in MakeNode()
831 Status GraphConstructor::ModifyNodeDefForImport(NodeDef* node_def) { in ModifyNodeDefForImport()
842 void RemoveInputs(const std::vector<int>& inputs_to_remove, NodeDef* node_def, in RemoveInputs()
864 NodeDef* node_def, std::vector<bool>* input_already_exists) { in RemapNodeDefInputs()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/
Dsharding_util.cc90 const NodeDef& node_def, int num_cores_per_replica, bool add_metadata) { in ParseShardingFromDevice()
154 const NodeDef& node_def, bool add_metadata, const char* attribute) { in GetShardingFromNodeDefInternal()
176 const NodeDef& node_def, bool add_metadata) { in GetShardingFromNodeDef()
/external/tensorflow/tensorflow/lite/toco/logging/
Dconversion_log_util_test.cc44 tensorflow::NodeDef node_def; in TEST() local
94 tensorflow::NodeDef node_def; in TEST() local
192 tensorflow::NodeDef node_def; in TEST() local
/external/tensorflow/tensorflow/core/util/
Dmirror_pad_mode.cc24 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name, in GetNodeAttr()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dvariable_merger_pass.cc75 NodeDef node_def; in MergeVarHandleOps() local
103 NodeDef node_def; in MergeReadVariableOps() local
Dtpu_embedding_software_deduplication_rewrite_pass.cc697 std::vector<std::string> GetControlInputs(const NodeDef& node_def) { in GetControlInputs()
713 std::vector<NodeDefBuilder::NodeOut> GetDataInputs(const NodeDef& node_def, in GetDataInputs()
727 const NodeDef& node_def) { in ComputeRewriterConfigForNodeDef()
828 for (const NodeDef& node_def : fdef.node_def()) { in ComputeRewriterConfigsByDevice() local
906 for (const NodeDef& node_def : fdef->node_def()) { in RewriteFunctionDefs() local
/external/tensorflow/tensorflow/python/util/
Dkernel_registry.cc29 tensorflow::NodeDef node_def; in TryFindKernelClass() local

1234567