Home
last modified time | relevance | path

Searched defs:out_tensor (Results 1 – 25 of 47) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsummary_op_test.cc72 Tensor* out_tensor = GetOutput(0); in TEST_F() local
100 Tensor* out_tensor = GetOutput(0); in TEST_F() local
129 Tensor* out_tensor = GetOutput(0); in TEST_F() local
203 Tensor* out_tensor = GetOutput(0); in TEST_F() local
237 Tensor* out_tensor = GetOutput(0); in TEST_F() local
H A Dsummary_image_op_test.cc87 Tensor* out_tensor = GetOutput(0); in TEST_F() local
110 Tensor* out_tensor = GetOutput(0); in TEST_F() local
142 Tensor* out_tensor = GetOutput(0); in TEST_F() local
H A Dsummary_audio_op_test.cc93 Tensor* out_tensor = GetOutput(0); in TEST_F() local
127 Tensor* out_tensor = GetOutput(0); in TEST_F() local
159 Tensor* out_tensor = GetOutput(0); in TEST_F() local
H A Dtensor_list_util.cc60 Tensor out_tensor; in TensorListBinaryAdd() local
76 Tensor out_tensor; in TensorListZerosLike() local
H A Dsummary_tensor_op_test.cc80 Tensor* out_tensor = GetOutput(0); in TEST_F() local
H A Dcount_up_to_op.cc49 Tensor* out_tensor; in Compute() local
H A Dhistogram_op.cc119 Tensor* out_tensor; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dlight_outside_compilation_kernels_for_test.cc42 Tensor* out_tensor = nullptr; in Compute() local
140 Tensor* out_tensor = nullptr; in Compute() local
196 Tensor* out_tensor = nullptr; in Compute() local
289 Tensor* out_tensor = nullptr; in Compute() local
331 Tensor* out_tensor = nullptr; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dbuffer_map_test.cc119 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST() local
137 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST() local
189 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST() local
208 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST() local
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A Dcomm.cpp57 for (auto& out_tensor : out_tensors) { in _broadcast_out_impl() local
66 for (auto& out_tensor : out_tensors) { in _broadcast_out_impl() local
375 at::Tensor& out_tensor, in _broadcast_out_impl()
393 at::Tensor& out_tensor, in _broadcast_out_impl()
H A Dpython_comm.cpp102 int64_t dim) { return gather_out(tensors, out_tensor, dim); }, in initCommMethods()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dmessage_wrappers.cc27 Tensor* out_tensor) { in ParseTensorProtoToTensor()
619 Status InMemoryRunGraphResponse::RecvValue(size_t i, TensorProto* out_tensor) { in RecvValue()
624 Status InMemoryRunGraphResponse::RecvValue(size_t i, Tensor* out_tensor) { in RecvValue()
682 TensorProto* out_tensor) { in RecvValue()
687 Status OwnedProtoRunGraphResponse::RecvValue(size_t i, Tensor* out_tensor) { in RecvValue()
757 TensorProto* out_tensor) { in RecvValue()
762 Status NonOwnedProtoRunGraphResponse::RecvValue(size_t i, Tensor* out_tensor) { in RecvValue()
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/utils/
H A DDependencyGraph.h200 for(auto out_tensor : outputs) variable
379 void link_output(OperatorId op, TensorId out_tensor) in link_output()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/python/
H A Dtoco_from_protos_test.py33 def _run(self, sess, in_tensor, out_tensor, should_succeed): argument
/aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/
H A Ddiag_test.py39 def forward(self, input, diagonal: int, out: bool, out_tensor): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/
H A Dinterpreter_utils.cc70 const TfLiteTensor* out_tensor = interpreter->tensor(t); in InterpreterInvokeWithOpResolver() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Doptional_ops_util.cc71 Tensor out_tensor; in OptionalBinaryAdd() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dcheckpoint_reader.cc89 const string& name, std::unique_ptr<tensorflow::Tensor>* out_tensor, in GetTensor()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DTensorTransformations.cu119 auto out_tensor = at::empty_like(in_tensor, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in roll_cuda() local
/aosp_15_r20/external/executorch/backends/xnnpack/runtime/
H A DXNNExecutor.cpp192 Tensor* out_tensor = &args[ext_id]->toTensor(); in resize_outputs() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_split_with_sizes_copy.cpp80 const Tensor& out_tensor = out[i]; in split_with_sizes_copy_out() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_transpose_op.cc66 Tensor* out_tensor, const gtl::ArraySlice<int32>& perm) { in MKLTransposeND()
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/internal/
H A Dtfprof_show.cc59 std::unique_ptr<Tensor> out_tensor; in LookUpCheckPoint() local
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DStaging.cpp25 const ValueRef out_tensor) { in add_staging_to_tensor_node()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_matmul_acl.h37 arm_compute::Tensor out_tensor; member

12