Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_batch_normalization.cc95 auto& mean_shape = *mean_array.mutable_shape(); in Run() local
98 if (mean_shape.dims().empty()) { in Run()
99 *mean_shape.mutable_dims() = multiplier_shape.dims(); in Run()
102 data.resize(RequiredBufferSizeForShape(mean_shape), data[0]); in Run()
104 CHECK(mean_shape.dims() == multiplier_shape.dims()); in Run()
105 CHECK(mean_shape.dims() == offset_shape.dims()); in Run()
106 const auto& param_shape = mean_shape; in Run()
/external/tensorflow/tensorflow/python/ops/
Drandom_grad.py192 mean_shape = array_ops.shape(mean)
198 mean_shape, stddev_shape)
212 broadcast_shape, mean_shape)
221 math_ops.reduce_sum(grad_mean, axis=rmean, keepdims=True), mean_shape)
/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.cc1317 const Shape& offset_shape, const Shape& mean_shape, in InferBatchNormInferenceShape() argument
1329 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(mean_shape)); in InferBatchNormInferenceShape()
1398 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(mean_shape, in InferBatchNormInferenceShape()
1405 PrimitiveType_Name(mean_shape.element_type()), in InferBatchNormInferenceShape()
1416 PrimitiveType_Name(mean_shape.element_type()), in InferBatchNormInferenceShape()
1440 if (ShapeUtil::GetDimension(mean_shape, 0) != feature_count) { in InferBatchNormInferenceShape()
1445 ShapeUtil::GetDimension(mean_shape, 0), feature_count); in InferBatchNormInferenceShape()
1461 const Shape& mean_shape, const Shape& var_shape, in InferBatchNormGradShape() argument
1466 TF_RETURN_IF_ERROR(ExpectArray(mean_shape, "mean input of batch norm grad")); in InferBatchNormGradShape()
1472 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.cc2672 TF_ASSIGN_OR_RETURN(const Shape* mean_shape, GetShapePtr(mean)); in BatchNormInference()
2677 *mean_shape, *variance_shape, feature_index)); in BatchNormInference()