Searched defs:mean_shape (Results 1 – 3 of 3) sorted by relevance
93 auto& mean_shape = *mean_array.mutable_shape(); in Run() local
94 Shape mean_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
1273 const Shape& offset_shape, const Shape& mean_shape, in InferBatchNormInferenceShape()1417 const Shape& mean_shape, const Shape& var_shape, in InferBatchNormGradShape()