Searched refs:tensor_out_arg_max (Results 1 – 2 of 2) sorted by relevance
254 Tensor tensor_out_arg_max; in Compute() local257 &tensor_out_arg_max)); in Compute()266 tensor_out_arg_max.flat<int64>().data(), output_size[3], in Compute()269 tensor_out_arg_max.flat<int64>().setConstant(kInvalidMaxPoolingIndex); in Compute()344 auto out_arg_max_flat = tensor_out_arg_max.flat<int64>(); in Compute()
266 Tensor tensor_out_arg_max; in Compute() local269 &tensor_out_arg_max)); in Compute()309 context, &tensor_out_dup, &tensor_out_arg_max, output, tensor_in, in Compute()