Home
last modified time | relevance | path

Searched refs:GetGridSize (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dwinograd.h41 int3 GetGridSize() const override;
70 int3 GetGridSize() const override;
111 int3 GetGridSize() const override;
137 int3 GetGridSize() const override;
Dresize.h32 int3 GetGridSize() const override;
58 int3 GetGridSize() const override;
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
Dcumsum.h34 int3 GetGridSize() const override;
Dstrided_slice.h32 int3 GetGridSize() const override;
Dconv_weights_converter.h35 int3 GetGridSize() const override;
Dsplit.h32 int3 GetGridSize() const override;
Dsoftmax1x1.h38 int3 GetGridSize() const override;
Dmean_stddev_normalization.h46 int3 GetGridSize() const override;
Dreduce.h42 int3 GetGridSize() const override;
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
Dconvolution_transposed_3x3_thin.h37 int3 GetGridSize() const override;
Dconv_metal_simd.h38 int3 GetGridSize() const override;
Dconvolution_transposed_4x4.h47 int3 GetGridSize() const override;
Dconvolution_transposed_3x3.h45 int3 GetGridSize() const override;
Dsoftmax1x1.cc145 int3 Softmax1x1::GetGridSize() const { in GetGridSize() function in tflite::gpu::Softmax1x1
Dcumsum.cc87 int3 Cumsum::GetGridSize() const { in GetGridSize() function in tflite::gpu::Cumsum
Dstrided_slice.cc170 int3 StridedSlice::GetGridSize() const { in GetGridSize() function in tflite::gpu::StridedSlice
Ddepthwise_conv_3x3.h45 int3 GetGridSize() const override;
Ddepthwise_conv_3x3_stride_h2.h44 int3 GetGridSize() const override;
Dconvolution_transposed_thin.h41 int3 GetGridSize() const override;
Dconvolution_transposed.h49 int3 GetGridSize() const override;
Dconvolution_transposed_thin.cc150 int3 ConvolutionTransposedThin::GetGridSize() const { in GetGridSize() function in tflite::gpu::ConvolutionTransposedThin
Dfully_connected.h126 int3 GetGridSize() const override;
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dgpu_operation.h170 void RecalculateGridSize() { grid_size_ = GetGridSize(); } in RecalculateGridSize()
205 virtual int3 GetGridSize() const;

12