Home
last modified time | relevance | path

Searched defs:input_split (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/graph/mutators/
H A DGroupedConvolutionMutator.cpp53 …NodeID input_split = GraphBuilder::add_split_node(g, params, input, num_grou… in create_grouped_convolution() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator.cc264 std::vector<absl::string_view> input_split = in IsEinsumCorrectlyFormed() local
1154 std::vector<absl::string_view> input_split = in GenerateBatchMatmulContextFromEinsum() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc6263 dnn::BatchDescriptor input_split = input_dimensions; in DoPoolForward() local
6332 dnn::BatchDescriptor input_split = input_dimensions; in DoPoolBackward() local