Home
last modified time | relevance | path

Searched refs:tuple_input (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/api/
H A Dmodule.cpp246 auto tuple_input = c10::ivalue::Tuple::create(inputs); in operator ()() local
247 IValue result = Method(_ivalue(), pre_hook)({tuple_input}); in operator ()()
262 auto tuple_input = c10::ivalue::Tuple::create(inputs); in operator ()() local
263 auto hook_result = Method(_ivalue(), hook)({tuple_input, outputs}); in operator ()()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf_control_flow.cc273 Value tuple_input = builder->create<mhlo::TupleOp>(loc, implicit_inputs_ref); in TupleImplicitInputs() local
279 block.addArgument(tuple_input.getType(), loc); in TupleImplicitInputs()
283 return tuple_input; in TupleImplicitInputs()
/aosp_15_r20/external/tensorflow/tensorflow/python/trackable/
H A Ddata_structures_test.py712 def f(tuple_input): argument
713 return tuple_input[0] + constant_op.constant(1.)