Searched defs:source_shape (Results 1 – 8 of 8) sorted by relevance
46 const Shape& source_shape = in ExecuteOnStream() local
2062 const Shape source_shape = GetShape(select_and_scatter_op.source()); in EmitSelectAndScatter() 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
317 absl::optional<HloSharding> ReshapeSharding(const Shape& source_shape, in ReshapeSharding()
2359 const Window& window, const Shape& source_shape, in InferSelectAndScatterShape()
122 const Shape& source_shape, const Shape& dest_shape, in StrideConfig()