Searched defs:input_tuple (Results 1 – 7 of 7) sorted by relevance
22 /*static*/ HloInstruction* TupleUtil::ExtractPrefix(HloInstruction* input_tuple, in ExtractPrefix()42 HloInstruction* input_tuple, in AppendSuffix()
47 static HloInstruction* Duplicate(HloInstruction* input_tuple) { in Duplicate()
321 auto input_tuple = in InvertDiagonalBlocks() local
297 auto input_tuple = xla::Tuple(b, inputs); in Compile() local
318 xla::XlaOp input_tuple = xla::Tuple(b, inputs); in Compile() local
3868 auto input_tuple = loop->mutable_operand(0); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() local4150 auto* input_tuple = loop->mutable_operand(0); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
2532 tensorflow::Safe_PyObjectPtr input_tuple(PySequence_Tuple(inputs)); local