Searched defs:offset_shape (Results 1 – 3 of 3) sorted by relevance
85 const auto& offset_shape = offset_array.shape(); in Run() local
45 Shape offset_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
1181 const Shape& offset_shape, int64 feature_index) { in InferBatchNormTrainingShape()1286 const Shape& offset_shape, const Shape& mean_shape, in InferBatchNormInferenceShape()