Searched refs:MeanStdDevNormalization (Results 1 – 4 of 4) sorted by relevance
34 class MeanStdDevNormalization : public GPUOperation {36 explicit MeanStdDevNormalization(const OperationDef& definition,49 MeanStdDevNormalization(MeanStdDevNormalization&& kernel) = default;50 MeanStdDevNormalization& operator=(MeanStdDevNormalization&& kernel) =52 MeanStdDevNormalization(const MeanStdDevNormalization&) = delete;53 MeanStdDevNormalization& operator=(const MeanStdDevNormalization&) = delete;62 MeanStdDevNormalization CreateMeanStdDevNormalization(
52 std::make_unique<MeanStdDevNormalization>(std::move(operation)), in MeanStddevNormSeparateBatchesTest()74 std::make_unique<MeanStdDevNormalization>( in MeanStddevNormSeparateBatchesTest()113 std::make_unique<MeanStdDevNormalization>(std::move(operation)), in MeanStddevNormalizationAllBatchesTest()138 std::make_unique<MeanStdDevNormalization>( in MeanStddevNormalizationAllBatchesTest()184 std::make_unique<MeanStdDevNormalization>(std::move(operation)), in MeanStddevNormalizationLargeVectorTest()211 std::make_unique<MeanStdDevNormalization>( in MeanStddevNormalizationLargeVectorTest()
132 MeanStdDevNormalization::MeanStdDevNormalization(const OperationDef& definition, in MeanStdDevNormalization() function in tflite::gpu::MeanStdDevNormalization216 std::string MeanStdDevNormalization::GetNormalizationCode( in GetNormalizationCode()323 int3 MeanStdDevNormalization::GetGridSize() const { in GetGridSize()332 MeanStdDevNormalization CreateMeanStdDevNormalization( in CreateMeanStdDevNormalization()335 return MeanStdDevNormalization(definition, gpu_info, shape, variance_bias, in CreateMeanStdDevNormalization()430 std::make_unique<MeanStdDevNormalization>(CreateMeanStdDevNormalization( in TryMeanStdDevNormalization()
598 MeanStdDevNormalization operation = CreateMeanStdDevNormalization( in GPUOperationFromNodePart0()600 *gpu_op = std::make_unique<MeanStdDevNormalization>(std::move(operation)); in GPUOperationFromNodePart0()