Home
last modified time | relevance | path

Searched defs:tensor_in_shape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_common.cc50 const TensorShape& tensor_in_shape) { in PoolParameters()
290 const TensorShape& tensor_in_shape, bool propagate_nans) { in Compute()
Davgpooling_op.cc243 const Tensor& tensor_in_shape = context->input(0); in Compute() local
402 const Tensor& tensor_in_shape = context->input(0); in Compute() local
482 const Tensor& tensor_in_shape = context->input(0); in Compute() local
Dmkl_avgpooling_op.cc245 const Tensor& tensor_in_shape = MklGetInput(context, 0); in Compute() local
353 const Tensor& tensor_in_shape = MklGetInput(context, 0); in MklCreateLayoutsAndPrimitives() local
673 const Tensor& tensor_in_shape, in SanityCheckInputs()
Dmkl_pooling_ops_common.cc228 const TensorShape& tensor_in_shape) { in Init()
Dcudnn_pooling_gpu.cc123 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute()
Dpooling_ops_3d.cc57 const TensorShape& tensor_in_shape) { in Pool3dParameters()
494 const Tensor& tensor_in_shape = context->input(0); in Compute() local
Dmaxpooling_op.cc326 const Tensor& out_backprop, const TensorShape& tensor_in_shape) { in MaxPoolingBackwardCustomKernel()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc308 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile() local
515 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile() local