Searched refs:tensor_out_slice (Results 1 – 1 of 1) sorted by relevance
270 Eigen::Tensor<T, 5, Eigen::RowMajor> tensor_out_slice(src_sizes); in launch() local271 tensor_out_slice.device(context->eigen_cpu_device()) = in launch()285 ((tensor_in_slice - tensor_out_slice.broadcast(bcast)).abs() < in launch()