Searched refs:output_tensor_base (Results 1 – 1 of 1) sorted by relevance
55 auto& output_tensor_base = output_operand.tensor_base(); in abs_kernel() local66 TORCH_CHECK(input_tensor_base.sizes() == output_tensor_base.sizes(), in abs_kernel()103 at::Tensor output = at::empty(output_tensor_base.sizes(), in abs_kernel()