Home
last modified time | relevance | path

Searched defs:mean_shape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dresolve_batch_normalization.cc95 auto& mean_shape = *mean_array.mutable_shape(); in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbatchnorm_expander_test.cc93 Shape mean_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
H A Dshape_inference.cc1316 const Shape& offset_shape, const Shape& mean_shape, in InferBatchNormInferenceShape()
1460 const Shape& mean_shape, const Shape& var_shape, in InferBatchNormGradShape()