Searched defs:infeed_shape (Results 1 – 6 of 6) sorted by relevance
121 const auto infeed_shape = ShapeUtil::MakeShape(F32, {3}); in TEST_F() local223 const auto build_body = [&result_shape](const Shape& infeed_shape) { in TEST_F()
40 const auto& infeed_shape = infeed_slices_.shape(); in ExecuteOnStream() local
233 if (absl::optional<Shape> infeed_shape = GetXfeedShape( in ReplayComputation() local
975 const Shape& infeed_shape() const { in infeed_shape() function
863 const Shape& infeed_shape, HloInstruction* token_operand, in CreateInfeed()
1772 HloInfeedInstruction::HloInfeedInstruction(const Shape& infeed_shape, in HloInfeedInstruction()