Searched refs:mean_shape (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
D | resolve_batch_normalization.cc | 95 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/ |
D | random_grad.py | 192 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/ |
D | batchnorm_expander_test.cc | 94 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()
|
D | shape_inference.cc | 1317 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 …]
|
D | shape_inference.h | 97 const Shape& offset_shape, const Shape& mean_shape, 103 const Shape& mean_shape,
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 2672 TF_ASSIGN_OR_RETURN(const Shape* mean_shape, GetShapePtr(mean)); in BatchNormInference() 2677 *mean_shape, *variance_shape, feature_index)); in BatchNormInference()
|