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.h40 int3 GetGridSize() const override;
67 int3 GetGridSize() const override;
107 int3 GetGridSize() const override;
133 int3 GetGridSize() const override;
Dresize.h30 int3 GetGridSize() const override;
56 int3 GetGridSize() const override;
Dsplit.h29 int3 GetGridSize() const override;
Dconv_weights_converter.h32 int3 GetGridSize() const override;
Dstrided_slice.h30 int3 GetGridSize() const override;
Dmean_stddev_normalization.h40 int3 GetGridSize() const override;
Dsoftmax1x1.h36 int3 GetGridSize() const override;
Dwinograd.cc189 int3 Winograd4x4To36::GetGridSize() const { in GetGridSize() function in tflite::gpu::Winograd4x4To36
460 int3 Winograd4x4To36TileX6::GetGridSize() const { in GetGridSize() function in tflite::gpu::Winograd4x4To36TileX6
494 int3 Winograd36To4x4::GetGridSize() const { in GetGridSize() function in tflite::gpu::Winograd36To4x4
672 int3 Winograd36To4x4Tile4x1::GetGridSize() const { in GetGridSize() function in tflite::gpu::Winograd36To4x4Tile4x1
Dreduce.h41 int3 GetGridSize() const override;
Dconvolution_transposed_3x3_thin.h40 int3 GetGridSize() const override;
Dresize.cc146 int3 Resize::GetGridSize() const { in GetGridSize() function in tflite::gpu::Resize
303 int3 Resize3D::GetGridSize() const { in GetGridSize() function in tflite::gpu::Resize3D
Dconvolution_transposed_4x4.h47 int3 GetGridSize() const override;
Dconvolution_transposed_3x3.h45 int3 GetGridSize() const override;
Dconv_metal.h63 int3 GetGridSize() const override;
Dsoftmax1x1.cc137 int3 Softmax1x1::GetGridSize() const { in GetGridSize() function in tflite::gpu::Softmax1x1
Dmean_stddev_normalization.cc208 int3 MeanStdDevNormalization::GetGridSize() const { in GetGridSize() function in tflite::gpu::MeanStdDevNormalization
Dfully_connected.cc153 int3 FullyConnected::GetGridSize() const { in GetGridSize() function in tflite::gpu::FullyConnected
Dsplit.cc139 int3 Split::GetGridSize() const { in GetGridSize() function in tflite::gpu::Split
Dconv_weights_converter.cc178 int3 ConverterToConvWeights::GetGridSize() const { in GetGridSize() function in tflite::gpu::ConverterToConvWeights
Dconvolution_transposed_thin.h38 int3 GetGridSize() const override;
Dfully_connected.h125 int3 GetGridSize() const override;
Dstrided_slice.cc169 int3 StridedSlice::GetGridSize() const { in GetGridSize() function in tflite::gpu::StridedSlice
Ddepthwise_conv_3x3.h43 int3 GetGridSize() const override;
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.cc171 int3 FCFCAdd::GetGridSize() const { return int3(dst_[0]->Slices(), 1, 1); } in GetGridSize() function in tflite::gpu::FCFCAdd
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dgpu_operation.h165 virtual int3 GetGridSize() const;

12