Searched defs:Infeed (Results 1 – 5 of 5) sorted by relevance
1758 ## Infeed section in Operation Semantics
91 TEST_F(XlaBuilderTest, Infeed) { in TEST_F() argument
958 struct Infeed { struct961 static Infeed Handler() { return Infeed(); } in Handler()1012 static bool Infeed(runtime::KernelContext* ctx, void** args, void** attrs) { in Infeed() function
247 TEST_F(HloControlFlowFlatteningTest, Infeed) { in TEST_F() argument
1748 XlaOp XlaBuilder::Infeed(const Shape& shape, const std::string& config) { in Infeed() function in xla::XlaBuilder4405 XlaOp Infeed(XlaBuilder* builder, const Shape& shape, in Infeed() function