Searched defs:tensor_out (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_input_conversion_op.cc | 110 Tensor* tensor_out; in Compute() local
211 Tensor* tensor_out; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | maxpooling_op.cc | 268 const Tensor& tensor_out = context->input(1); in Compute() local
394 const Tensor& tensor_out = context->input(1); in Compute() local
501 const Tensor& tensor_out = context->input(1); in Compute() local
571 const Tensor& tensor_in, const Tensor& tensor_out, in SpatialMaxPoolGradGrad()
715 const Tensor& tensor_out = context->input(1); in Compute() local
|
| H A D | pooling_ops_common.cc | 224 Tensor* tensor_out = nullptr; in Compute() local
444 TensorFormat data_format, const Tensor* tensor_in, const Tensor* tensor_out, in Compute()
|
| H A D | pooling_ops_3d.cc | 338 const Tensor& tensor_out = context->input(1); in Compute() local
704 const Tensor& tensor_out = context->input(1); in Compute() local
|
| H A D | cudnn_pooling_gpu.cc | 134 const Tensor* tensor_in, const Tensor* tensor_out, Tensor* input_backprop) { in Compute()
|
| H A D | fractional_max_pool_op.cc | 240 const Tensor& tensor_out = context->input(1); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | batch_matmul.cc | 393 const scalar* input, TfLiteTensor* tensor_out, in TransposeRowsColumnsImpl()
413 TfLiteTensor* tensor_out) { in TransposeRowsColumns()
|
| /aosp_15_r20/external/executorch/backends/arm/runtime/ |
| H A D | ArmBackendEthosU.cpp | 246 auto tensor_out = args[handles.inputs->count + i]->toTensor(); in execute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | mkl_util.h | 789 inline void AllocTmpBuffer(OpKernelContext* context, Tensor* tensor_out, in AllocTmpBuffer()
800 inline void AllocTmpBuffer(OpKernelContext* context, Tensor* tensor_out, in AllocTmpBuffer()
|
| /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
|