/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | tuple_util.cc | 22 /*static*/ HloInstruction* TupleUtil::ExtractPrefix(HloInstruction* input_tuple, in ExtractPrefix() argument 24 CHECK(input_tuple->shape().IsTuple()); in ExtractPrefix() 26 HloComputation* computation = input_tuple->parent(); in ExtractPrefix() 27 const Shape& input_shape = input_tuple->shape(); in ExtractPrefix() 34 input_shape.tuple_shapes(i), input_tuple, i))); in ExtractPrefix() 42 HloInstruction* input_tuple, in AppendSuffix() argument 44 CHECK(input_tuple->shape().IsTuple()); in AppendSuffix() 46 HloComputation* computation = input_tuple->parent(); in AppendSuffix() 47 const Shape& input_shape = input_tuple->shape(); in AppendSuffix() 53 input_shape.tuple_shapes(i), input_tuple, i))); in AppendSuffix()
|
H A D | tuple_util.h | 31 static HloInstruction* ExtractPrefix(HloInstruction* input_tuple, 41 HloInstruction* input_tuple, 47 static HloInstruction* Duplicate(HloInstruction* input_tuple) { in Duplicate() argument 48 return ExtractPrefix(input_tuple, input_tuple->shape().tuple_shapes_size()); in Duplicate()
|
H A D | triangular_solve_expander.cc | 321 auto input_tuple = in InvertDiagonalBlocks() local 324 auto i = GetTupleElement(input_tuple, 0); in InvertDiagonalBlocks() 325 auto body_out = GetTupleElement(input_tuple, 1); in InvertDiagonalBlocks() 326 auto body_input = GetTupleElement(input_tuple, 2); in InvertDiagonalBlocks()
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | common_nn.py | 3630 def _check_gradients(self, test_case, module, input_tuple): argument 3632 num_inputs = len(input_tuple) 3644 test_input_jacobian = torch.is_floating_point(input_tuple[0]) 3645 test_case.check_jacobian(module, input_tuple[0], test_input_jacobian) 3647 test_case.assertTrue(gradcheck(fn_to_gradcheck, input_tuple + params, 3653 test_case.assertTrue(gradgradcheck(fn_to_gradcheck, input_tuple + params, 3661 input_tuple = input if isinstance(input, tuple) else (input,) 3663 self._check_gradients(test_case, module, input_tuple) 3674 assert len(input_tuple) == 1 3675 input = input_tuple[0] [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | if_op.cc | 318 xla::XlaOp input_tuple = xla::Tuple(b, inputs); in Compile() local 320 ctx->builder(), ctx->Input(0), input_tuple, *then_result.computation, in Compile() 321 input_tuple, *else_result.computation); in Compile()
|
H A D | case_op.cc | 297 auto input_tuple = xla::Tuple(b, inputs); in Compile() local 300 std::vector<xla::XlaOp>(num_branches, input_tuple)); in Compile()
|
/aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
H A D | python_function.cpp | 856 THPObjectPtr input_tuple; member 875 unpacked.input_tuple = PyTuple_New(num_args); in unpack_input() 914 PyTuple_SET_ITEM(unpacked.input_tuple.get(), i, arg); in unpack_input() 1227 PyObject* arg = PyTuple_GET_ITEM(unpacked_input.input_tuple.get(), i); in make_ctx_input_tuple() 1242 Py_INCREF(unpacked_input.input_tuple.get()); in make_ctx_input_output_tuple() 1245 PyTuple_SET_ITEM(result.get(), 1, unpacked_input.input_tuple.get()); in make_ctx_input_output_tuple() 1351 output = PyObject_CallObject(forward_fn, unpacked_input.input_tuple); in THPFunction_apply()
|
/aosp_15_r20/packages/modules/Bluetooth/system/gd/rust/linux/dbus_projection/dbus_macros/src/ |
D | lib.rs | 438 let input_tuple = quote! { in generate_dbus_interface_client() localVariable 447 self.client_proxy.method_noreturn(#dbus_method_name, #input_tuple) in generate_dbus_interface_client() 454 #input_tuple, in generate_dbus_interface_client() 465 .async_method_noreturn(#dbus_method_name, #input_tuple) in generate_dbus_interface_client() 472 .async_method(#dbus_method_name, #input_tuple) in generate_dbus_interface_client()
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
H A D | script_init.cpp | 1152 const py::tuple& input_tuple, in initJitScriptBindings() 1160 auto typed_inputs = toTraceableStack(input_tuple); in initJitScriptBindings() 1321 const py::tuple& input_tuple) { in initJitScriptBindings() argument 1323 for (auto& input : input_tuple) { in initJitScriptBindings() 1332 [](mobile::Module& m, const py::tuple& input_tuple) { in initJitScriptBindings() argument 1334 for (auto& input : input_tuple) { in initJitScriptBindings() 1647 const py::tuple& input_tuple, in initJitScriptBindings() 1652 auto typed_inputs = toTraceableStack(input_tuple); in initJitScriptBindings()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
H A D | padding_fifo_queue_test.py | 1569 input_tuple = [] 1579 input_tuple.append(np_array) 1581 q.enqueue_many(input_tuple).run() 1586 for (input_elem, output_elem) in zip(input_tuple, output_tuple):
|
H A D | fifo_queue_test.py | 592 input_tuple = [] 602 input_tuple.append(np_array) 604 q.enqueue_many(input_tuple).run() 609 for (input_elem, output_elem) in zip(input_tuple, output_tuple):
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
H A D | unpack_quantized_weights.cpp | 581 auto input_tuple = in UnpackQuantizedTensorInputs() local 583 g_input->replaceAllUsesWith(input_tuple); in UnpackQuantizedTensorInputs()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
H A D | dot_handler.cc | 3868 auto input_tuple = loop->mutable_operand(0); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() local 3869 auto old_operand = input_tuple->mutable_operand(non_windowed_operand_index); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() 3880 TF_RETURN_IF_ERROR(input_tuple->ReplaceOperandWithDifferentShape( in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() 3888 input_tuple, loop, loop->while_condition()->parameter_instruction(0), in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() 4150 auto* input_tuple = loop->mutable_operand(0); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local 4156 input_tuple->ReplaceOperandWithDifferentShape(2, new_input_subtuple)); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() 4163 input_tuple, loop, loop->while_condition()->parameter_instruction(0), in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
|
/aosp_15_r20/external/pytorch/docs/source/ |
H A D | jit.rst | 483 for input_tuple in [inputs] + check_inputs: 484 torch.testing.assert_close(fn(*input_tuple), scripted_fn(*input_tuple))
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_view_ops.py | 1591 input_tuple = (s, a, b, c, d) 1592 gradcheck(lambda s, w, x, y, z: torch_fn(s, w, x, y, z), input_tuple) 1593 gradgradcheck(lambda s, w, x, y, z: torch_fn(s, w, x, y, z), input_tuple)
|
H A D | test_mps.py | 10322 input_tuple = (s, a, b, c, d) 10323 gradcheck(lambda s, w, x, y, z: torch_fn(s, w, x, y, z), input_tuple) 10324 gradgradcheck(lambda s, w, x, y, z: torch_fn(s, w, x, y, z), input_tuple)
|
H A D | test_nn.py | 106 input_tuple = input if isinstance(input, tuple) else (input,) 108 for i in input_tuple: 111 args = input_tuple + (target,) + extra_args
|
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
H A D | pywrap_tfe_src.cc | 2532 tensorflow::Safe_PyObjectPtr input_tuple(PySequence_Tuple(inputs)); local 2535 Py_BuildValue("OOOOOO", op_name, attrs, input_tuple.get(), results,
|
/aosp_15_r20/external/pytorch/torch/_C/ |
H A D | __init__.pyi.in | 494 input_tuple: Tuple[Any, ...],
|