Searched defs:input_tuple (Results 1 – 12 of 12) 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
1157 bool store_inputs) { in initJitScriptBindings()1321 const py::tuple& input_tuple) { in initJitScriptBindings()1332 [](mobile::Module& m, const py::tuple& input_tuple) { in initJitScriptBindings()1651 const std::vector<std::string>& argument_names) { in initJitScriptBindings()
581 auto input_tuple = in UnpackQuantizedTensorInputs() local
438 let input_tuple = quote! { in generate_dbus_interface_client() localVariable
3630 def _check_gradients(self, test_case, module, input_tuple): argument
3868 auto input_tuple = loop->mutable_operand(0); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() local4150 auto* input_tuple = loop->mutable_operand(0); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
856 THPObjectPtr input_tuple; member
2532 tensorflow::Safe_PyObjectPtr input_tuple(PySequence_Tuple(inputs)); local