Searched defs:offset_shape (Results 1 – 5 of 5) sorted by relevance
97 const auto& offset_shape = offset_array.shape(); in Run() local
84 offset_shape: Optional[torch.Size] = None variable in _MaskPartial
58 Shape offset_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
1211 const Shape& offset_shape, int64_t feature_index) { in InferBatchNormTrainingShape()1316 const Shape& offset_shape, const Shape& mean_shape, in InferBatchNormInferenceShape()
3150 const Shape& offset_shape = per_partition_size_hlo->shape(); in HandleDynamicUpdateSlice() local