Searched defs:dynamic_slice_shape (Results 1 – 3 of 3) sorted by relevance
356 Shape dynamic_slice_shape = lhs.hlo()->shape(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
340 Shape dynamic_slice_shape = ShapeUtil::MakeShape(F32, {4, 4}); in TEST_F() local
1987 const Shape& dynamic_slice_shape = a->shape(); in OptimizeDotOfGather() local