Home
last modified time | relevance | path

Searched refs:mean_shape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_batch_normalization.cc93 auto& mean_shape = *mean_array.mutable_shape(); in Run() local
96 if (mean_shape.dims().empty()) { in Run()
97 *mean_shape.mutable_dims() = multiplier_shape.dims(); in Run()
100 data.resize(RequiredBufferSizeForShape(mean_shape), data[0]); in Run()
102 CHECK(mean_shape.dims() == multiplier_shape.dims()); in Run()
103 CHECK(mean_shape.dims() == offset_shape.dims()); in Run()
104 const auto& param_shape = mean_shape; in Run()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander_test.cc94 Shape mean_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
106 HloInstruction::CreateParameter(2, mean_shape, "mean")); in TEST_F()
115 ShapeUtil::MakeTupleShape({input_shape, scale_shape, mean_shape}), param0, in TEST_F()
Dshape_inference.cc1273 const Shape& offset_shape, const Shape& mean_shape, in InferBatchNormInferenceShape() argument
1285 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(mean_shape)); in InferBatchNormInferenceShape()
1354 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(mean_shape, in InferBatchNormInferenceShape()
1361 PrimitiveType_Name(mean_shape.element_type()), in InferBatchNormInferenceShape()
1372 PrimitiveType_Name(mean_shape.element_type()), in InferBatchNormInferenceShape()
1396 if (ShapeUtil::GetDimension(mean_shape, 0) != feature_count) { in InferBatchNormInferenceShape()
1401 ShapeUtil::GetDimension(mean_shape, 0), feature_count); in InferBatchNormInferenceShape()
1417 const Shape& mean_shape, const Shape& var_shape, in InferBatchNormGradShape() argument
1422 TF_RETURN_IF_ERROR(ExpectArray(mean_shape, "mean input of batch norm grad")); in InferBatchNormGradShape()
1428 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(mean_shape)); in InferBatchNormGradShape()
[all …]
Dshape_inference.h97 const Shape& offset_shape, const Shape& mean_shape,
103 const Shape& mean_shape,
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2078 TF_ASSIGN_OR_RETURN(const Shape& mean_shape, GetShape(mean)); in BatchNormInference()
2082 operand_shape, scale_shape, offset_shape, mean_shape, in BatchNormInference()