Searched defs:tensor_out (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | mkl_input_conversion_op.cc | 123 Tensor* tensor_out; in Compute() local 221 Tensor* tensor_out; in Compute() local
|
D | pooling_ops_common.cc | 161 Tensor* tensor_out = nullptr; in Compute() local 289 const Tensor* tensor_out, const Tensor& out_backprop, in Compute()
|
D | maxpooling_op.cc | 248 const Tensor& tensor_out = context->input(1); in Compute() local 380 const Tensor& tensor_out = context->input(1); in Compute() local 486 const Tensor& tensor_out = context->input(1); in Compute() local 539 const Tensor& tensor_in, const Tensor& tensor_out, in SpatialMaxPoolGradGrad() 683 const Tensor& tensor_out = context->input(1); in Compute() local
|
D | cudnn_pooling_gpu.cc | 124 const Tensor* tensor_in, const Tensor* tensor_out, Tensor* input_backprop) { in Compute()
|
D | pooling_ops_3d.cc | 341 const Tensor& tensor_out = context->input(1); in Compute() local 685 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/core/util/ |
D | mkl_util.h | 1040 inline void AllocTmpBuffer(OpKernelContext* context, Tensor* tensor_out, in AllocTmpBuffer() 1050 inline void AllocTmpBuffer(OpKernelContext* context, Tensor* tensor_out, in AllocTmpBuffer() 1065 inline void AllocTmpBuffer(OpKernelContext* context, Tensor* tensor_out, in AllocTmpBuffer()
|