Home
last modified time | relevance | path

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

1234567

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A 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 …]
H A 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
H A Ddevice.py119 def __call__(self, node_def): argument
128 def shortcut_string_merge(self, node_def): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A 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 …]
H A 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 …]
H A 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()
H A 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
H A Dop_kernel_test.cc151 NodeDef node_def; in CreateNodeDef() local
184 NodeDef node_def; in ExpectFailure() local
330 NodeDef node_def = CreateNodeDef("Test1", {DT_FLOAT, DT_INT32}); in TEST_F() local
339 NodeDef node_def = CreateNodeDef("Test1", {DT_FLOAT, DT_INT32}); in TEST_F() local
903 const NodeDef& node_def, const char* input_name, in BM_InputRangeHelper()
928 NodeDef node_def; in BM_ConcatInputRange() local
949 NodeDef node_def; in BM_SelectInputRange() local
966 NodeDef node_def; in BM_TraceString() local
H A Dnode_properties_test.cc62 const NodeDef& node_def, in ValidateNodeProperties()
83 NodeDef node_def; in TEST() local
105 NodeDef node_def; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A 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 …]
H A 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 …]
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A 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 …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
H A 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
H A 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
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper.cc582 const auto* node_def = node_view.node(); in IsBiasSemanticAdd() local
666 const auto* node_def = node_view->node(); in FindContractionWithBias() local
709 const auto* node_def = node_view->node(); in FindContractionWithBiasAndActivation() local
760 const auto* node_def = node_view->node(); in FindConvWithSqueezeAndBias() local
814 const auto* node_def = node_view->node(); in FindConv2DWithBatchNorm() local
872 const auto* node_def = node_view->node(); in FindConv2DWithBatchNormAndActivation() local
944 const auto* node_def = node_view->node(); in FindPadWithConv3D() local
982 const auto* node_def = node_view.node(); in FindContractionWithBiasAddAndAdd() local
1028 const auto* node_def = node_view->node(); in FindContractionWithBiasAndAddActivation() local
1087 NodeDef* node_def = node_view->node(); in VerifyConstants() local
[all …]
H A Dgeneric_layout_optimizer.cc79 const auto* node_def = node.node(); in NumConvOnDeviceWithDataTypeOverThreshold() local
132 auto* node_def = node_view->node(); in ExpandLayoutSensitiveOp() local
154 auto* node_def = node_view->node(); in ExpandLayoutAgnosticOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/logging/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_test_util.cc312 bool IsPlaceholder(const tensorflow::NodeDef& node_def) { in IsPlaceholder()
332 bool IsScalarConst(const tensorflow::NodeDef& node_def, int v) { in IsScalarConst()
358 bool IsAddN(const tensorflow::NodeDef& node_def, int n) { in IsAddN()
383 bool IsNeg(const tensorflow::NodeDef& node_def, const string& input) { in IsNeg()
400 bool GetNodeDef(TF_Operation* oper, tensorflow::NodeDef* node_def) { in GetNodeDef()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dexecute.cc268 const auto& node_def = op->MutableAttrs()->BuildNodeDef(); in GetOutputDTypes() local
296 const KernelDef* GetKernelDef(const EagerOperation& op, const NodeDef* node_def, in GetKernelDef()
307 bool IsHostMemoryArg(const EagerOperation& op, const NodeDef* node_def, in IsHostMemoryArg()
452 const auto& node_def = op->MutableAttrs()->BuildNodeDef(); in VerifyWrappableInCallOp() local
531 const NodeDef& node_def = op->MutableAttrs()->BuildNodeDef(); in ValidateOp() local
970 const NodeDef* node_def = nullptr; in ExtractFunctionInputInfo() local
1082 const NodeDef* node_def = &op->MutableAttrs()->BuildNodeDef(); in GetOrCreateKernelAndDevice() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dextract_outside_compilation_pass_test.cc525 for (const auto &node_def : true_def->node_def()) { in TEST_F() local
537 for (const auto &node_def : false_def->node_def()) { in TEST_F() local
667 for (const auto &node_def : cond_def->node_def()) { in TEST_F() local
679 for (const auto &node_def : body_def->node_def()) { in TEST_F() local
695 for (const auto &node_def : cond_def->node_def()) { in TEST_F() local

1234567