Searched defs:tensor_in (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| 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 …]
|
| 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()
|
| 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()
|
| 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
|
| D | quantized_pooling_ops.cc | 58 const Tensor& tensor_in = context->input(0); in Compute() local
|
| 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
|
| 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()
|
| 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
|
| D | fractional_avg_pool_op.cc | 76 const Tensor& tensor_in = context->input(0); in Compute() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_matmul.cc | 392 void TransposeRowsColumnsImpl(const TfLiteTensor* tensor_in, in TransposeRowsColumnsImpl()
412 const TfLiteTensor* tensor_in, in TransposeRowsColumns()
|