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()
46 static HloInstruction* Duplicate(HloInstruction* input_tuple) { in Duplicate()
328 auto input_tuple = in InvertDiagonalBlocks() local
294 xla::XlaOp input_tuple = xla::Tuple(b, inputs); in Compile() local
305 auto input_tuple = xla::Tuple(b, inputs); in Compile() local
2932 auto input_tuple = loop->mutable_operand(0); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() local3171 auto input_tuple = loop->mutable_operand(0); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
2460 tensorflow::Safe_PyObjectPtr input_tuple(PySequence_Tuple(inputs)); local