Home
last modified time | relevance | path

Searched defs:tensor_out (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_input_conversion_op.cc123 Tensor* tensor_out; in Compute() local
221 Tensor* tensor_out; in Compute() local
Dpooling_ops_common.cc161 Tensor* tensor_out = nullptr; in Compute() local
289 const Tensor* tensor_out, const Tensor& out_backprop, in Compute()
Dmaxpooling_op.cc248 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
Dcudnn_pooling_gpu.cc124 const Tensor* tensor_in, const Tensor* tensor_out, Tensor* input_backprop) { in Compute()
Dpooling_ops_3d.cc341 const Tensor& tensor_out = context->input(1); in Compute() local
685 const Tensor& tensor_out = context->input(1); in Compute() local
Dfractional_max_pool_op.cc231 const Tensor& tensor_out = context->input(1); in Compute() local
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1040 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()