Home
last modified time | relevance | path

Searched defs:output_list (Results 1 – 25 of 26) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dlist_kernels.cc99 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 Dlist_kernels.h374 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 Dscoped_allocator_ops_test.cc181 OpOutputList output_list; in ExecOp() local
303 OpOutputList output_list; in ExecOp() local
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dcustom_function.h287 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 Dbackend.cpp20 c10::List<at::Tensor> output_list; in execute() local
/aosp_15_r20/external/pytorch/test/custom_backend/
H A Dcustom_backend.h40 c10::List<at::Tensor> output_list; in execute() local
/aosp_15_r20/external/executorch/test/
H A Dsize_test.cpp88 auto output_list = in main() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/xnnpack/
H A Dxnnpack_backend_preprocess.cpp105 auto output_list = out.toList(); in preprocess() local
H A Dxnnpack_backend_lib.cpp105 c10::List<at::Tensor> output_list(output_tensors); in execute() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_squeeze_op_test.py179 def test_passing_ragged(self, input_list, output_list, squeeze_ranks=None): argument
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_backend_lib.cpp42 c10::List<at::Tensor> output_list; in execute() local
H A Dtest_backend_compiler_lib.cpp114 c10::List<at::Tensor> output_list; in execute() local
H A Dtest_flatbuffer.cpp1703 auto output_list = output.toTupleRef().elements(); in TEST() local
H A Dtest_lite_interpreter.cpp1966 auto output_list = output.toTupleRef().elements(); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dunified_api_testutil.cc77 OutputList output_list; in RunModel() local
/aosp_15_r20/external/federated-compute/fcp/tensorflow/
H A Dtf_session.cc101 std::vector<tensorflow::Tensor> output_list; in GetOutputs() local
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/experimental/
H A Dunified_api.cc147 OutputList output_list; in PYBIND11_MODULE() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dfunction_cache_test.cc132 tensorflow::OutputList output_list; in PrepareFunction() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dvararg_functions.cpp420 auto output_list = c10::impl::GenericList(elems.elementType()); in dequantize() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dtensor_list_utils.cc133 xla::XlaOp* output_list) { in BuildNonNestedTensorList()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_c_api_client.cc479 auto& output_list = ret.emplace_back(); in Convert2DCBuffersToCppBuffers() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dgraph_fuser.cpp593 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 DProcessGroupGloo.cpp2120 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 Dtfe_wrapper.cc349 PyObject* output_list = PyList_New(output_len); in InputTFE_OutputTensorHandles() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel.cc685 Status OpKernelContext::output_list(StringPiece name, OpOutputList* list) { in output_list() function in tensorflow::OpKernelContext

12