| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | maxpooling_op.cc | 65 Tensor* input_backprop, const Tensor& tensor_in, const Tensor& out_backprop, in SpatialMaxPoolWithArgMaxHelper()
267 const Tensor& tensor_in = context->input(0); in Compute() local
393 const Tensor& tensor_in = context->input(0); in Compute() local
500 const Tensor& tensor_in = context->input(0); in Compute() local
571 const Tensor& tensor_in, const Tensor& tensor_out, in SpatialMaxPoolGradGrad()
714 const Tensor& tensor_in = context->input(0); in Compute() local
833 const Tensor& tensor_in = context->input(0); in Compute() local
893 const Tensor& tensor_in = context->input(0); in Compute() local
984 const Tensor& tensor_in = context->input(0); in Compute() local
1116 const Tensor& tensor_in = context->input(0); in Compute() local
[all …]
|
| H A D | pooling_ops_common.h | 130 const Tensor& tensor_in = context->input(0); in Compute() local
180 const Tensor& tensor_in, const PoolParameters& params) { in DepthwiseMaxPool()
190 const Tensor& tensor_in, const PoolParameters& params, in SpatialMaxPool()
|
| H A D | cudnn_pooling_gpu.cc | 41 const Tensor& tensor_in, Tensor* output) { in Compute()
134 const Tensor* tensor_in, const Tensor* tensor_out, Tensor* input_backprop) { in Compute()
|
| H A D | pooling_ops_3d.cc | 167 const Tensor& tensor_in = context->input(0); in Compute() local
337 const Tensor& tensor_in = context->input(0); in Compute() local
703 const Tensor& tensor_in = context->input(0); in Compute() local
|
| H A D | fractional_max_pool_op.cc | 79 const Tensor& tensor_in = context->input(0); in Compute() local
239 const Tensor& tensor_in = context->input(0); in Compute() local
|
| H A D | pooling_ops_common.cc | 221 TensorFormat data_format, const Tensor& tensor_in, in Compute()
444 TensorFormat data_format, const Tensor* tensor_in, const Tensor* tensor_out, in Compute()
|
| H A D | avgpooling_op.cc | 89 const Tensor& tensor_in = context->input(0); in Compute() local
169 const Tensor& tensor_in = context->input(0); in Compute() local
|
| H A D | quantized_pooling_ops.cc | 58 const Tensor& tensor_in = context->input(0); in Compute() local
|
| H A D | fractional_avg_pool_op.cc | 76 const Tensor& tensor_in = context->input(0); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | batch_matmul.cc | 392 void TransposeRowsColumnsImpl(const TfLiteTensor* tensor_in, in TransposeRowsColumnsImpl()
412 const TfLiteTensor* tensor_in, in TransposeRowsColumns()
|
| /aosp_15_r20/external/executorch/backends/arm/runtime/ |
| H A D | ArmBackendEthosU.cpp | 141 auto tensor_in = args[i]->toTensor(); in execute() local
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | distributed_test.py | 2528 self, tensor_out, tensor_in, group_id, rank, cuda=True, rank_to_GPU=None argument
3549 self, tensor_out, tensor_in, group_id, rank, cuda=True, rank_to_GPU=None argument
|