Home
last modified time | relevance | path

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

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