Searched defs:tensor_in (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op.cc | 64 Tensor* input_backprop, const Tensor& tensor_in, const Tensor& out_backprop, in SpatialMaxPoolWithArgMaxHelper()
263 const Tensor& tensor_in = context->input(0); in Compute() local
381 const Tensor& tensor_in = context->input(0); in Compute() local
488 const Tensor& tensor_in = context->input(0); in Compute() local
547 const Tensor& tensor_in, const Tensor& tensor_out, in SpatialMaxPoolGradGrad()
690 const Tensor& tensor_in = context->input(0); in Compute() local
798 const Tensor& tensor_in = context->input(0); in Compute() local
858 const Tensor& tensor_in = context->input(0); in Compute() local
949 const Tensor& tensor_in = context->input(0); in Compute() local
1067 const Tensor& tensor_in = context->input(0); in Compute() local
[all …]
|
| D | pooling_ops_common.h | 135 const Tensor& tensor_in = context->input(0); in Compute() local
185 const Tensor& tensor_in, const PoolParameters& params) { in DepthwiseMaxPool()
195 const Tensor& tensor_in, const PoolParameters& params, in SpatialMaxPool()
|
| D | cudnn_pooling_gpu.cc | 40 const Tensor& tensor_in, Tensor* output) { in Compute()
139 const Tensor* tensor_in, const Tensor* tensor_out, Tensor* input_backprop) { in Compute()
|
| D | quantized_pooling_ops.cc | 56 const Tensor& tensor_in = context->input(0); in Compute() local
|
| D | fractional_max_pool_op.cc | 78 const Tensor& tensor_in = context->input(0); in Compute() local
230 const Tensor& tensor_in = context->input(0); in Compute() local
|
| D | pooling_ops_3d.cc | 162 const Tensor& tensor_in = context->input(0); in Compute() local
336 const Tensor& tensor_in = context->input(0); in Compute() local
680 const Tensor& tensor_in = context->input(0); in Compute() local
|
| D | avgpooling_op.cc | 88 const Tensor& tensor_in = context->input(0); in Compute() local
158 const Tensor& tensor_in = context->input(0); in Compute() local
|
| D | pooling_ops_common.cc | 218 TensorFormat data_format, const Tensor& tensor_in, in Compute()
445 TensorFormat data_format, const Tensor* tensor_in, const Tensor* tensor_out, in Compute()
|
| D | fractional_avg_pool_op.cc | 75 const Tensor& tensor_in = context->input(0); in Compute() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_matmul.cc | 388 void TransposeRowsColumnsImpl(const TfLiteTensor* tensor_in, in TransposeRowsColumnsImpl()
408 const TfLiteTensor* tensor_in, in TransposeRowsColumns()
|