Home
last modified time | relevance | path

Searched defs:tensor_out_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_avgpooling_op.cc124 TensorShape tensor_out_shape; in Compute() local
314 TensorShape tensor_out_shape; in Compute() local
Dpooling_ops_common.cc159 const TensorShape& tensor_out_shape, in Compute()
Dmkl_maxpooling_op.cc105 TensorShape tensor_out_shape; in Compute() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc309 const TensorShape tensor_out_shape = ctx->InputShape(1); in Compile() local
516 const TensorShape tensor_out_shape = ctx->InputShape(1); in Compile() local