Home
last modified time | relevance | path

Searched refs:input_tuple (Results 1 – 19 of 19) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtuple_util.cc22 /*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 Dtuple_util.h31 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 Dtriangular_solve_expander.cc321 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 Dcommon_nn.py3630 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 Dif_op.cc318 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 Dcase_op.cc297 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 Dpython_function.cpp856 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/
Dlib.rs438 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 Dscript_init.cpp1152 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 Dpadding_fifo_queue_test.py1569 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 Dfifo_queue_test.py592 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 Dunpack_quantized_weights.cpp581 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 Ddot_handler.cc3868 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 Djit.rst483 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 Dtest_view_ops.py1591 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 Dtest_mps.py10322 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 Dtest_nn.py106 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 Dpywrap_tfe_src.cc2532 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.in494 input_tuple: Tuple[Any, ...],