Home
last modified time | relevance | path

Searched defs:out_tensors (Results 1 – 25 of 104) sorted by relevance

12345

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dlower_case_op_test.cc138 std::vector<Tensor> out_tensors; in TEST() local
147 std::vector<Tensor> out_tensors; in TEST() local
156 std::vector<Tensor> out_tensors; in TEST() local
165 std::vector<Tensor> out_tensors; in TEST() local
174 std::vector<Tensor> out_tensors; in TEST() local
278 std::vector<Tensor> out_tensors; in TEST() local
288 std::vector<Tensor> out_tensors; in TEST() local
298 std::vector<Tensor> out_tensors; in TEST() local
308 std::vector<Tensor> out_tensors; in TEST() local
318 std::vector<Tensor> out_tensors; in TEST() local
[all …]
H A Dlower_if_op_test.cc134 std::vector<Tensor> out_tensors; in TEST() local
143 std::vector<Tensor> out_tensors; in TEST() local
242 std::vector<Tensor> out_tensors; in TEST() local
252 std::vector<Tensor> out_tensors; in TEST() local
324 std::vector<Tensor> out_tensors; in TEST() local
333 std::vector<Tensor> out_tensors; in TEST() local
H A Dlower_while_op_test.cc159 std::vector<Tensor> out_tensors; in TEST() local
167 std::vector<Tensor> out_tensors; in TEST() local
495 std::vector<Tensor> out_tensors; in TEST() local
506 std::vector<Tensor> out_tensors; in TEST() local
575 std::vector<Tensor> out_tensors; in TEST() local
583 std::vector<Tensor> out_tensors; in TEST() local
H A Dlower_functional_ops_test.cc186 std::vector<Tensor> out_tensors; in TEST() local
196 std::vector<Tensor> out_tensors; in TEST() local
312 std::vector<Tensor> out_tensors; in TEST() local
321 std::vector<Tensor> out_tensors; in TEST() local
H A Dlower_function_call_op_test.cc129 std::vector<Tensor> out_tensors; in TEST() local
194 std::vector<Tensor> out_tensors; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dcsv_dataset_op.cc288 std::vector<Tensor>* out_tensors, in GetNextInternal()
401 Status ReadRecord(IteratorContext* ctx, std::vector<Tensor>* out_tensors, in ReadRecord()
446 std::vector<Tensor>* out_tensors, in ParseOneField()
509 std::vector<Tensor>* out_tensors, in ParseQuotedField()
580 std::vector<Tensor>* out_tensors, in QuotedFieldToOutput()
643 std::vector<Tensor>* out_tensors, in ParseUnquotedField()
711 std::vector<Tensor>* out_tensors) { in FieldToOutput()
830 std::vector<Tensor>* out_tensors, in UnquotedFieldToOutput()
H A Dsnapshot_dataset_op.cc313 std::vector<Tensor>* out_tensors, in GetNextInternal()
375 std::vector<Tensor>* out_tensors, in GetNextInternal()
575 std::vector<Tensor>* out_tensors, in GetNextInternal()
614 std::vector<Tensor>* out_tensors, in GetNextInternal()
1130 std::vector<Tensor>* out_tensors, in GetNextInternal()
1300 std::vector<Tensor>* out_tensors, in GetNextInternal()
1692 std::vector<Tensor>* out_tensors, in GetNextInternal()
2225 std::vector<Tensor>* out_tensors, in GetNextInternal()
H A Dchoose_fastest_branch_dataset_op.cc100 std::vector<Tensor>* out_tensors, in GetNextInternal()
361 std::vector<Tensor>* out_tensors, in GetNextInternal()
457 std::vector<Tensor>* out_tensors, in GetNextFromExperiment()
H A Dlmdb_dataset_op_test.cc171 std::vector<Tensor> out_tensors; in TEST_F() local
185 std::vector<Tensor> out_tensors; in TEST_F() local
H A Dsql_dataset_op.cc144 std::vector<Tensor>* out_tensors, in GetNextInternal()
184 std::vector<Tensor> out_tensors; in RestoreInternal() local
H A Dthreadpool_dataset_op.cc238 std::vector<Tensor>* out_tensors, in GetNextInternal()
358 std::vector<Tensor>* out_tensors, in GetNextInternal()
498 std::vector<Tensor>* out_tensors, in GetNextInternal()
H A Dio_ops_test.cc165 std::vector<Tensor> out_tensors; in TEST_P() local
264 std::vector<Tensor> out_tensors; in TEST_P() local
H A Dlist_dataset_op_test.cc205 std::vector<Tensor> out_tensors; in TEST_P() local
370 std::vector<Tensor> out_tensors; in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dcache_dataset_ops.cc82 std::vector<Tensor>* out_tensors) { in Get()
102 std::vector<Tensor> out_tensors; in ExtendTempCacheToIndex() local
220 std::vector<Tensor>* out_tensors, in GetNextInternal()
320 std::vector<Tensor>* out_tensors, in GetNextInternal()
561 std::vector<Tensor>* out_tensors, in GetNextInternal()
808 std::vector<Tensor>* out_tensors, in GetNextInternal()
867 std::vector<Tensor>* out_tensors, in GetNextInternal()
944 std::vector<Tensor>* out_tensors, in GetNextInternal()
H A Drepeat_dataset_op.cc144 std::vector<Tensor>* out_tensors, in GetNextInternal()
177 std::vector<Tensor>* out_tensors, in GetNextInternal()
252 std::vector<Tensor>* out_tensors, in GetNextInternal()
H A Dtensor_slice_dataset_op_test.cc111 std::vector<Tensor> out_tensors; in TEST_P() local
281 std::vector<Tensor> out_tensors; in TEST_P() local
H A Drepeat_dataset_op_test.cc144 std::vector<Tensor> out_tensors; in TEST_P() local
322 std::vector<Tensor> out_tensors; in TEST_P() local
H A Dcache_dataset_ops_test.cc178 std::vector<Tensor> out_tensors; in TEST_P() local
319 std::vector<Tensor> out_tensors; in TEST_P() local
H A Dtake_dataset_op.cc111 Status GetNextInternal(IteratorContext* ctx, std::vector<Tensor>* out_tensors, in GetNextInternal()
144 Status GetNextInternal(IteratorContext* ctx, std::vector<Tensor>* out_tensors, in GetNextInternal()
H A Dskip_dataset_op.cc119 std::vector<Tensor>* out_tensors, in GetNextInternal()
153 std::vector<Tensor>* out_tensors, in GetNextInternal()
H A Dtensor_dataset_op_test.cc130 std::vector<Tensor> out_tensors; in TEST_P() local
236 std::vector<Tensor> out_tensors; in TEST_P() local
H A Dsparse_tensor_slice_dataset_op_test.cc173 std::vector<Tensor> out_tensors; in TEST_P() local
381 std::vector<Tensor> out_tensors; in TEST_P() local
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A Dcomm.cpp52 std::vector<Tensor>& out_tensors) { in _broadcast_out_impl()
75 std::vector<Tensor>& out_tensors) { in _broadcast_out_impl()
233 std::vector<at::Tensor>& out_tensors, in _broadcast_out_impl()
H A Dpython_comm.cpp39 [](at::Tensor& tensor, std::vector<at::Tensor>& out_tensors) { in initCommMethods()
72 std::optional<py::object> py_streams) { in initCommMethods()
/aosp_15_r20/external/tensorflow/tensorflow/examples/label_image/
H A Dmain.cc119 std::vector<Tensor>* out_tensors) { in ReadTensorFromImageFile()
223 std::vector<Tensor> out_tensors; in GetTopLabels() local

12345