Home
last modified time | relevance | path

Searched defs:concat_node (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dsparsify_gather_test.cc150 NodeDef* concat_node = nullptr; in TestSinglePartition() local
383 NodeDef* concat_node = CreateNode( in TestMultiPartition() local
H A Dbackports_test.cc75 NodeDef* concat_node = graph_def.add_node(); in TestBackportConcatV2() local
H A Dbackports.cc38 NodeDef concat_node = concat_v2_node; in BackportConcatV2Transform() local
H A Dfold_old_batch_norms.cc231 NodeDef concat_node = concat_node_match.node; in FuseBatchNormWithConvConcat() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
H A Dmake_padding_test.cc40 auto concat_node = graph.NewNode(); in TEST() local
/aosp_15_r20/external/ComputeLibrary/src/graph/mutators/
H A DDepthConcatSubTensorMutator.cpp71 … auto *concat_node = arm_compute::utils::cast::polymorphic_downcast<ConcatenateLayerNode *>(node); in mutate() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dpeephole.cpp656 auto concat_node = transformToONNXConcatNode( in eraseListConstruct() local
974 auto concat_node = *it; in removeSequenceSplitConcat() local
H A Dpreprocess_for_onnx.cpp151 Node* concat_node = b->owningGraph()->create(onnx::Concat, 1); in ReplaceAddWithConcat() local
H A Dhelper.cpp232 Node* concat_node = need_new_input in transformToONNXConcatNode() local
H A Dshape_type_inference.cpp273 auto concat_node = transformToONNXConcatNode( in CloneValueFromListConstruct() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
H A Dconv_pointwise.cc236 NodeContext concat_node; in TryFusedPointwiseConv() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_functional_ops.cc791 Node* concat_node = nullptr; in CreateConcatAndSplitNodesForInputTensor() local
967 Node* concat_node = nullptr; in CreateConcatAndSplitNodesForOutputTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc1172 Node* concat_node, in CreateSliceNode()
3779 Node* concat_node = nullptr; in BuildExecuteNodes() local
3870 Node* concat_node = nullptr; in BuildExecuteNodes() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dmodel_builder.cc749 Node* concat_node = graph->NewNode(); in ResolveGroupedConvolution() local
1381 Node* concat_node = graph->NewNode(); in ParseBasic() local
/aosp_15_r20/external/openscreen/third_party/abseil/src/absl/strings/
H A Dcord.cc1107 CordRepConcat* concat_node = node->concat(); in Build() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer.cc1768 NodeDef* concat_node) { in HoistChainForConcat()