Home
last modified time | relevance | path

Searched refs:tensor_out_dup (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfractional_max_pool_op.cc273 Tensor tensor_out_dup; in Compute() local
276 &tensor_out_dup)); in Compute()
286 tensor_out_dup.flat<T>().data(), output_size[3], in Compute()
294 tensor_out_dup.flat<T>().setConstant(Eigen::NumTraits<T>::lowest()); in Compute()
H A Dmaxpooling_op.cc282 Tensor tensor_out_dup; in Compute() local
285 &tensor_out_dup)); in Compute()
342 context, &tensor_out_dup, &tensor_out_arg_max, output, tensor_in, in Compute()