Home
last modified time | relevance | path

Searched defs:GetGridSize (Results 1 – 22 of 22) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dwinograd.cc227 int3 Winograd4x4To36::GetGridSize() const { in GetGridSize() function in tflite::gpu::Winograd4x4To36
506 int3 Winograd4x4To36TileX6::GetGridSize() const { in GetGridSize() function in tflite::gpu::Winograd4x4To36TileX6
540 int3 Winograd36To4x4::GetGridSize() const { in GetGridSize() function in tflite::gpu::Winograd36To4x4
722 int3 Winograd36To4x4Tile4x1::GetGridSize() const { in GetGridSize() function in tflite::gpu::Winograd36To4x4Tile4x1
H A Dresize.cc101 int3 Resize::GetGridSize() const { in GetGridSize() function in tflite::gpu::Resize
200 int3 Resize3D::GetGridSize() const { in GetGridSize() function in tflite::gpu::Resize3D
H A Dsoftmax1x1.cc145 int3 Softmax1x1::GetGridSize() const { in GetGridSize() function in tflite::gpu::Softmax1x1
H A Dcumsum.cc87 int3 Cumsum::GetGridSize() const { in GetGridSize() function in tflite::gpu::Cumsum
H A Dconv_weights_converter.cc230 int3 ConverterToConvWeights::GetGridSize() const { in GetGridSize() function in tflite::gpu::ConverterToConvWeights
H A Dstrided_slice.cc170 int3 StridedSlice::GetGridSize() const { in GetGridSize() function in tflite::gpu::StridedSlice
H A Ddepthwise_conv_3x3_stride_h2.cc196 int3 DepthWiseConv3x3StrideH2::GetGridSize() const { in GetGridSize() function in tflite::gpu::DepthWiseConv3x3StrideH2
H A Dconvolution_transposed_thin.cc150 int3 ConvolutionTransposedThin::GetGridSize() const { in GetGridSize() function in tflite::gpu::ConvolutionTransposedThin
H A Ddepthwise_conv_3x3.cc298 int3 DepthwiseConv3x3::GetGridSize() const { in GetGridSize() function in tflite::gpu::DepthwiseConv3x3
H A Dfully_connected.cc193 int3 FullyConnected::GetGridSize() const { in GetGridSize() function in tflite::gpu::FullyConnected
H A Dsplit.cc199 int3 Split::GetGridSize() const { in GetGridSize() function in tflite::gpu::Split
H A Dconvolution_transposed_3x3_thin.cc207 int3 ConvolutionTransposed3x3Thin::GetGridSize() const { in GetGridSize() function in tflite::gpu::ConvolutionTransposed3x3Thin
H A Dconvolution_transposed_4x4.cc360 int3 ConvolutionTransposed4x4::GetGridSize() const { in GetGridSize() function in tflite::gpu::ConvolutionTransposed4x4
H A Dmean_stddev_normalization.cc323 int3 MeanStdDevNormalization::GetGridSize() const { in GetGridSize() function in tflite::gpu::MeanStdDevNormalization
H A Dconvolution_transposed_3x3.cc358 int3 ConvolutionTransposed3x3::GetGridSize() const { in GetGridSize() function in tflite::gpu::ConvolutionTransposed3x3
H A Ddepthwise_conv.cc123 int3 DepthwiseConv::GetGridSize() const { in GetGridSize() function in tflite::gpu::DepthwiseConv
H A Dreduce.cc501 int3 Reduce::GetGridSize() const { in GetGridSize() function in tflite::gpu::Reduce
H A Dconvolution_transposed.cc590 int3 ConvolutionTransposed::GetGridSize() const { in GetGridSize() function in tflite::gpu::ConvolutionTransposed
H A Dconv_metal_simd.cc441 int3 ConvolutionMetalSimd::GetGridSize() const { in GetGridSize() function in tflite::gpu::ConvolutionMetalSimd
H A Dconv_generic.cc336 int3 ConvGeneric::GetGridSize() const { in GetGridSize() function in tflite::gpu::ConvGeneric
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
H A Dfc_fc_add.cc226 int3 FCFCAdd::GetGridSize() const { return int3(dst_[0]->Slices(), 1, 1); } in GetGridSize() function in tflite::gpu::FCFCAdd
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
H A Dgpu_operation.cc461 int3 GPUOperation::GetGridSize() const { in GetGridSize() function in tflite::gpu::GPUOperation