Home
last modified time | relevance | path

Searched defs:input_tuple (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_util.cc22 /*static*/ HloInstruction* TupleUtil::ExtractPrefix(HloInstruction* input_tuple, in ExtractPrefix()
42 HloInstruction* input_tuple, in AppendSuffix()
Dtuple_util.h46 static HloInstruction* Duplicate(HloInstruction* input_tuple) { in Duplicate()
Dtriangular_solve_expander.cc328 auto input_tuple = in InvertDiagonalBlocks() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dif_op.cc294 xla::XlaOp input_tuple = xla::Tuple(b, inputs); in Compile() local
Dcase_op.cc305 auto input_tuple = xla::Tuple(b, inputs); in Compile() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc2932 auto input_tuple = loop->mutable_operand(0); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() local
3171 auto input_tuple = loop->mutable_operand(0); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc2460 tensorflow::Safe_PyObjectPtr input_tuple(PySequence_Tuple(inputs)); local