Searched refs:MeanStdDevNormalization (Results 1 – 4 of 4) sorted by relevance
28 class MeanStdDevNormalization : public GPUOperation {30 explicit MeanStdDevNormalization(const OperationDef& definition,43 MeanStdDevNormalization(MeanStdDevNormalization&& kernel) = default;44 MeanStdDevNormalization& operator=(MeanStdDevNormalization&& kernel) =46 MeanStdDevNormalization(const MeanStdDevNormalization&) = delete;47 MeanStdDevNormalization& operator=(const MeanStdDevNormalization&) = delete;53 MeanStdDevNormalization CreateMeanStdDevNormalization(
90 MeanStdDevNormalization::MeanStdDevNormalization(const OperationDef& definition, in MeanStdDevNormalization() function in tflite::gpu::MeanStdDevNormalization151 std::string MeanStdDevNormalization::GetNormalizationCode( in GetNormalizationCode()208 int3 MeanStdDevNormalization::GetGridSize() const { in GetGridSize()217 MeanStdDevNormalization CreateMeanStdDevNormalization( in CreateMeanStdDevNormalization()220 return MeanStdDevNormalization(definition, gpu_info, tensor_slices); in CreateMeanStdDevNormalization()
49 absl::make_unique<MeanStdDevNormalization>(std::move(operation)), in MeanStddevNormSeparateBatchesTest()96 absl::make_unique<MeanStdDevNormalization>(std::move(operation)), in MeanStddevNormalizationAllBatchesTest()151 absl::make_unique<MeanStdDevNormalization>(std::move(operation)), in MeanStddevNormalizationLargeVectorTest()
436 MeanStdDevNormalization operation = CreateMeanStdDevNormalization( in GPUOperationFromNode()439 absl::make_unique<MeanStdDevNormalization>(std::move(operation)); in GPUOperationFromNode()