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
385 const Tensor& tensor_in = context->input(0); in Compute() local
492 const Tensor& tensor_in = context->input(0); in Compute() local
551 const Tensor& tensor_in, const Tensor& tensor_out, in SpatialMaxPoolGradGrad()
694 const Tensor& tensor_in = context->input(0); in Compute() local
802 const Tensor& tensor_in = context->input(0); in Compute() local
862 const Tensor& tensor_in = context->input(0); in Compute() local
953 const Tensor& tensor_in = context->input(0); in Compute() local
1085 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
693 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()
448 TensorFormat data_format, const Tensor* tensor_in, const Tensor* tensor_out, in Compute()
|
| D | avgpooling_op.cc | 88 const Tensor& tensor_in = context->input(0); in Compute() local
163 const Tensor& tensor_in = context->input(0); in Compute() local
|
| 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 | 389 void TransposeRowsColumnsImpl(const TfLiteTensor* tensor_in, in TransposeRowsColumnsImpl()
409 const TfLiteTensor* tensor_in, in TransposeRowsColumns()
|