Home
last modified time | relevance | path

Searched defs:split_node (Results 1 – 9 of 9) sorted by relevance

/external/ComputeLibrary/src/graph/mutators/
DSplitLayerSubTensorMutator.cpp83 … auto *split_node = arm_compute::utils::cast::polymorphic_downcast<SplitLayerNode *>(node); in mutate() local
/external/pytorch/torch/csrc/jit/passes/onnx/
Dfixup_onnx_controlflow.cpp146 Node* split_node = in ConvertSequenceDependencies() local
Dpeephole.cpp973 auto split_node = it->input()->node(); in removeSequenceSplitConcat() local
/external/pytorch/torch/_inductor/fx_passes/
Dsplit_cat.py99 def _get_split_args_default(split_node): argument
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_functional_ops.cc827 Node* split_node = nullptr; in CreateConcatAndSplitNodesForInputTensor() local
1011 Node* split_node = nullptr; in CreateConcatAndSplitNodesForOutputTensor() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc669 Node* split_node; member
680 Node* control_predecessor, Node* split_node, in CreatePadNode()
892 Node* split_node = split_nodes_for_dimension.front(); in CreateOrGetSplitNodesForInputSharding() local
/external/icing/icing/legacy/index/
Dicing-dynamic-trie.cc1650 Node *split_node = storage_->GetMutableNode(best_node_index); in Insert() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.cc689 Node* split_node = graph->NewNode(); in ResolveGroupedConvolution() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc1796 NodeDef* split_node) { in HoistChainForSplit()