| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | sharding_util_ops_test.cc | 54 std::vector<Tensor>* output_tensors) { in RunGraph()
87 std::vector<Tensor> output_tensors; in TEST() local
126 std::vector<Tensor> output_tensors; in TEST() local
223 std::vector<Tensor> output_tensors; in TEST_P() local
241 std::vector<Tensor> output_tensors; in TEST_P() local
259 std::vector<Tensor> output_tensors; in TEST_P() local
277 std::vector<Tensor> output_tensors; in TEST_P() local
294 std::vector<Tensor> output_tensors; in TEST_P() local
312 std::vector<Tensor> output_tensors; in TEST_P() local
329 std::vector<Tensor> output_tensors; in TEST_P() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | tape.py | 177 def record_operation(op_type, output_tensors, input_tensors, backward_function, argument
185 def record_operation_backprop_only(op_type, output_tensors, input_tensors, argument
193 def record_operation_forwardprop_only(op_type, output_tensors, input_tensors, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/ |
| H A D | lite.py | 1047 def _optimize_tf_model(self, graph_def, input_tensors, output_tensors, argument
1103 def convert(self, graph_def, input_tensors, output_tensors): argument
1994 def _optimize_tf_model(self, graph_def, input_tensors, output_tensors, argument
2380 output_tensors, argument
2561 output_tensors, argument
2592 def from_session(cls, sess, input_tensors, output_tensors): argument
2819 def from_session(cls, sess, input_tensors, output_tensors): argument
|
| H A D | util.py | 262 def _convert_op_hints_if_present(sess, graph_def, output_tensors, argument
273 def freeze_graph(sess, input_tensors, output_tensors): argument
|
| H A D | convert.py | 720 def convert_graphdef(input_data, input_tensors, output_tensors, **kwargs): argument
844 def toco_convert(input_data, input_tensors, output_tensors, *args, **kwargs): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | multi_threaded_pg.py | 351 def allgather(self, output_tensors, input_tensor, opts=AllgatherOptions()): argument
367 def scatter(self, output_tensors, input_tensors, opts=ScatterOptions()): argument
373 def gather(self, output_tensors, input_tensors, opts=ScatterOptions()): argument
389 …def reduce_scatter_tensor_coalesced(self, output_tensors, input_tensors, opts=ReduceScatterOptions… argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/testing/ |
| H A D | mlir_convert.py | 32 output_tensors, # pylint: disable=unused-argument argument
134 output_tensors, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/ |
| H A D | internal_api_samples.cc | 206 std::vector<std::vector<float>> output_tensors(out_refs.size()); in RunModelSampleWithInternalAPI() local
331 std::vector<std::vector<float>> output_tensors(out_refs.size()); in RunModelSampleWithInternalAPISerializedKernels() local
412 std::vector<std::vector<float>> output_tensors(out_refs.size()); in RunModelSampleWithInternalAPISerialized() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/mlir/testing/ |
| H A D | mlir_convert.py | 30 def mlir_convert(options, graph_def, input_tensors, output_tensors, **kwargs): argument
127 output_tensors, argument
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/vision/ |
| H A D | object_detector.cc | 163 const std::vector<const TfLiteTensor*>& output_tensors) { in SanityCheckOutputTensors()
447 const std::vector<const TfLiteTensor*>& output_tensors, in Postprocess()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| H A D | hexagon_delegate_kernel.cc | 125 std::vector<hexagon_nn_tensordef> output_tensors; in Eval() local
245 const TfLiteIntArray* output_tensors) { in BuildGraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/ |
| H A D | grappler_test.cc | 122 std::vector<Tensor> output_tensors; in EvaluateNodes() local
139 std::vector<Tensor> output_tensors; in EvaluateFetchNodes() local
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/ |
| H A D | nl_classifier.cc | 270 const std::vector<const TfLiteTensor*>& output_tensors, in Postprocess()
361 std::vector<const TfLiteTensor*> output_tensors = GetOutputTensors(); in Initialize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
| H A D | tape.h | 407 const string& op_type, const std::vector<TapeTensor>& output_tensors, in RecordOperation()
925 const string& op_type, const std::vector<TapeTensor>& output_tensors, in ForwardpropFromTape()
1024 const std::vector<TapeTensor>& output_tensors, in Accumulate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | io_ops.cc | 185 std::vector<Tensor> output_tensors; in GetShardIndex() local
352 std::vector<Tensor> output_tensors; in GetNextInternal() local
457 std::vector<Tensor> output_tensors; in GetShardIndex() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api_function.cc | 90 std::vector<OutputTensor>* output_tensors) in ProcessOutputs()
169 std::vector<tensorflow::OutputTensor> output_tensors; in TF_GraphToFunctionWithControlOutputs() local
|
| /aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/llama/runner/ |
| H A D | io_memory.cpp | 200 std::vector<std::vector<Tensor>>& output_tensors) { in update_io()
412 std::vector<std::vector<Tensor>>& output_tensors) { in update_io()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_fused_batch_norm_op_test.cc | 264 std::vector<Tensor> output_tensors; in VerifyFusedBatchNorm() local
401 std::vector<Tensor> output_tensors; in VerifyFusedBatchNormGradWithConv2D() local
|
| /aosp_15_r20/external/federated-compute/fcp/aggregation/protocol/ |
| H A D | configuration.proto | 58 repeated tensorflow.TensorSpecProto output_tensors = 5; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/benchmark/ |
| H A D | benchmark_model.cc | 126 std::vector<tensorflow::Tensor> output_tensors; in GetOutputShapes() local
295 std::vector<tensorflow::Tensor> output_tensors; in RunBenchmark() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/ |
| H A D | graph_info.h | 85 std::vector<int> output_tensors; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/distribute/experimental/rpc/proto/ |
| H A D | tf_rpc_service.proto | 14 repeated TensorProto output_tensors = 1; field
|
| /aosp_15_r20/external/federated-compute/fcp/client/engine/ |
| H A D | common.h | 54 std::vector<tensorflow::Tensor> output_tensors; member
|
| H A D | tflite_wrapper.h | 39 std::vector<tensorflow::Tensor> output_tensors; member
|
| /aosp_15_r20/external/federated-compute/fcp/aggregation/core/ |
| H A D | one_dim_grouping_aggregator.h | 74 OutputTensorList output_tensors = std::move(*other_ptr).TakeOutputs(); in MergeWith() local
|