Searched defs:tensor_out (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_input_conversion_op.cc | 122 Tensor* tensor_out; in Compute() local
223 Tensor* tensor_out; in Compute() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op.cc | 268 const Tensor& tensor_out = context->input(1); in Compute() local
386 const Tensor& tensor_out = context->input(1); in Compute() local
493 const Tensor& tensor_out = context->input(1); in Compute() local
551 const Tensor& tensor_in, const Tensor& tensor_out, in SpatialMaxPoolGradGrad()
695 const Tensor& tensor_out = context->input(1); in Compute() local
|
| D | pooling_ops_common.cc | 224 Tensor* tensor_out = nullptr; in Compute() local
448 TensorFormat data_format, const Tensor* tensor_in, const Tensor* tensor_out, in Compute()
|
| D | cudnn_pooling_gpu.cc | 139 const Tensor* tensor_in, const Tensor* tensor_out, Tensor* input_backprop) { in Compute()
|
| D | pooling_ops_3d.cc | 337 const Tensor& tensor_out = context->input(1); in Compute() local
694 const Tensor& tensor_out = context->input(1); in Compute() local
|
| D | fractional_max_pool_op.cc | 231 const Tensor& tensor_out = context->input(1); in Compute() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_matmul.cc | 390 const scalar* input, TfLiteTensor* tensor_out, in TransposeRowsColumnsImpl()
410 TfLiteTensor* tensor_out) { in TransposeRowsColumns()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 786 inline void AllocTmpBuffer(OpKernelContext* context, Tensor* tensor_out, in AllocTmpBuffer()
797 inline void AllocTmpBuffer(OpKernelContext* context, Tensor* tensor_out, in AllocTmpBuffer()
|