Home
last modified time | relevance | path

Searched refs:min_dimension (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClWidthConcatenate2TensorsKernel.cpp83 …const unsigned int min_dimension = std::min(src1->dimension(0), src2->dimensio… in configure() local
84 const unsigned int num_elems_processed_per_iteration = adjust_vec_size(8, min_dimension); in configure()
H A DClWidthConcatenate4TensorsKernel.cpp87 …const unsigned int min_dimension = std::min(std::min(src1->dimension(0), src2-… in configure() local
88 const unsigned int num_elems_processed_per_iteration = adjust_vec_size(8, min_dimension); in configure()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dragged_tensor_to_tensor_op.cc196 const INDEX_TYPE min_dimension = in CalculateFirstParentOutputIndex() local
200 for (INDEX_TYPE i = 0; i < min_dimension; in CalculateFirstParentOutputIndex()
204 for (INDEX_TYPE i = min_dimension; i < first_dimension; ++i) { in CalculateFirstParentOutputIndex()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
H A Dragged_tensor_to_tensor_tflite.cc299 const int min_dimension = std::min(first_dimension, first_dimension_output); in CalculateFirstParentOutputIndex() local
302 for (int i = 0; i < min_dimension; in CalculateFirstParentOutputIndex()
306 for (int i = min_dimension; i < first_dimension; ++i) { in CalculateFirstParentOutputIndex()