Searched defs:source_shape (Results 1 – 5 of 5) sorted by relevance
43 std::vector<int64> source_shape; member78 auto source_shape = GetParam().source_shape; in XLA_TEST_P() local
604 auto source_shape = ShapeUtil::MakeShape(F32, {2, 2, 2}); in TEST_F() local
476 const Shape& source_shape = operand->source()->shape(); in ReshapeToRemoveDegenerateDims() local
2126 const Window& window, const Shape& source_shape, in InferSelectAndScatterShape()
85 const Shape& source_shape, const Shape& dest_shape, in StrideConfig()