/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | list_kernels.cc | 99 TensorList** output_list) { in ForwardInputOrCreateNewList() 221 TensorList* output_list = nullptr; in Compute() local 402 TensorList output_list; in Compute() local 463 TensorList* output_list = nullptr; in Compute() local
|
H A D | list_kernels.h | 374 TensorList* output_list = nullptr; in Compute() local 596 TensorList output_list; in Compute() local 779 TensorList output_list; in Compute() local 874 TensorList* output_list = nullptr; in Compute() local 917 TensorList output_list; in Compute() local
|
H A D | scoped_allocator_ops_test.cc | 181 OpOutputList output_list; in ExecOp() local 303 OpOutputList output_list; in ExecOp() local
|
/aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
H A D | custom_function.h | 287 std::vector<std::optional<Variable>>& output_list) { in to_output_type() 299 std::vector<std::optional<Variable>>& output_list) { in to_output_type()
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/coreml/cpp/ |
H A D | backend.cpp | 20 c10::List<at::Tensor> output_list; in execute() local
|
/aosp_15_r20/external/pytorch/test/custom_backend/ |
H A D | custom_backend.h | 40 c10::List<at::Tensor> output_list; in execute() local
|
/aosp_15_r20/external/executorch/test/ |
H A D | size_test.cpp | 88 auto output_list = in main() local
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/xnnpack/ |
H A D | xnnpack_backend_preprocess.cpp | 105 auto output_list = out.toList(); in preprocess() local
|
H A D | xnnpack_backend_lib.cpp | 105 c10::List<at::Tensor> output_list(output_tensors); in execute() local
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
H A D | ragged_squeeze_op_test.py | 179 def test_passing_ragged(self, input_list, output_list, squeeze_ranks=None): argument
|
/aosp_15_r20/external/pytorch/test/cpp/jit/ |
H A D | test_backend_lib.cpp | 42 c10::List<at::Tensor> output_list; in execute() local
|
H A D | test_backend_compiler_lib.cpp | 114 c10::List<at::Tensor> output_list; in execute() local
|
H A D | test_flatbuffer.cpp | 1703 auto output_list = output.toTupleRef().elements(); in TEST() local
|
H A D | test_lite_interpreter.cpp | 1966 auto output_list = output.toTupleRef().elements(); in TEST() local
|
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
H A D | unified_api_testutil.cc | 77 OutputList output_list; in RunModel() local
|
/aosp_15_r20/external/federated-compute/fcp/tensorflow/ |
H A D | tf_session.cc | 101 std::vector<tensorflow::Tensor> output_list; in GetOutputs() local
|
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/experimental/ |
H A D | unified_api.cc | 147 OutputList output_list; in PYBIND11_MODULE() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/ |
H A D | function_cache_test.cc | 132 tensorflow::OutputList output_list; in PrepareFunction() local
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
H A D | vararg_functions.cpp | 420 auto output_list = c10::impl::GenericList(elems.elementType()); in dequantize() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | tensor_list_utils.cc | 133 xla::XlaOp* output_list) { in BuildNonNestedTensorList()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
H A D | pjrt_c_api_client.cc | 479 auto& output_list = ret.emplace_back(); in Convert2DCBuffersToCppBuffers() local
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
H A D | graph_fuser.cpp | 593 auto* output_list = g->insert(aten::broadcast_tensors, {input_list}); in broadcast_tensors() local
|
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
H A D | ProcessGroupGloo.cpp | 2120 for (auto& output_list : output_lists) { in allgather_coalesced() local 2158 for (const auto& output_list : output_lists) { in allgather_coalesced() local
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ |
H A D | tfe_wrapper.cc | 349 PyObject* output_list = PyList_New(output_len); in InputTFE_OutputTensorHandles() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
H A D | op_kernel.cc | 685 Status OpKernelContext::output_list(StringPiece name, OpOutputList* list) { in output_list() function in tensorflow::OpKernelContext
|