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