Searched defs:source_shape (Results 1 – 8 of 8) sorted by relevance
49 const Shape& source_shape = in ExecuteOnStream() local
2232 const Shape source_shape = in EmitSelectAndScatterFromMlir() local
43 std::vector<int64> source_shape; member78 auto source_shape = GetParam().source_shape; in XLA_TEST_P() local
476 const Shape& source_shape = operand->source()->shape(); in ReshapeToRemoveDegenerateDims() local
990 auto source_shape = ShapeUtil::MakeShape(F32, {2, 2, 2}); in TEST_F() local
297 absl::optional<HloSharding> ReshapeSharding(const Shape& source_shape, in ReshapeSharding()
2245 const Window& window, const Shape& source_shape, in InferSelectAndScatterShape()
119 const Shape& source_shape, const Shape& dest_shape, in StrideConfig()