Path Lines of Code torch_xla/csrc/aten_autograd_ops.cpp 128 torch_xla/csrc/aten_autograd_ops.h 31 torch_xla/csrc/aten_cpu_fallback.cpp 30 torch_xla/csrc/aten_cpu_fallback.h 5 torch_xla/csrc/aten_xla_bridge.cpp 276 torch_xla/csrc/aten_xla_bridge.h 51 torch_xla/csrc/aten_xla_type.cpp 3152 torch_xla/csrc/batch_norm.cpp 86 torch_xla/csrc/batch_norm.h 24 torch_xla/csrc/computation.cpp 9 torch_xla/csrc/computation.h 23 torch_xla/csrc/convert_ops.cpp 113 torch_xla/csrc/convert_ops.h 18 torch_xla/csrc/convolution.cpp 252 torch_xla/csrc/convolution.h 27 torch_xla/csrc/cross_replica_reduces.cpp 170 torch_xla/csrc/cross_replica_reduces.h 50 torch_xla/csrc/data_ops.cpp 436 torch_xla/csrc/data_ops.h 52 torch_xla/csrc/debug_util.cpp 122 torch_xla/csrc/debug_util.h 25 torch_xla/csrc/device.cpp 73 torch_xla/csrc/device.h 40 torch_xla/csrc/elementwise.cpp 233 torch_xla/csrc/elementwise.h 37 torch_xla/csrc/function_call_tracker.cpp 62 torch_xla/csrc/function_call_tracker.h 8 torch_xla/csrc/helpers.cpp 592 torch_xla/csrc/helpers.h 246 torch_xla/csrc/init_python_bindings.cpp 1107 torch_xla/csrc/ir.cpp 228 torch_xla/csrc/ir.h 180 torch_xla/csrc/ir_dump_util.cpp 230 torch_xla/csrc/ir_dump_util.h 20 torch_xla/csrc/ir_util.cpp 91 torch_xla/csrc/ir_util.h 28 torch_xla/csrc/layout_manager.cpp 161 torch_xla/csrc/layout_manager.h 14 torch_xla/csrc/lowering_context.cpp 162 torch_xla/csrc/lowering_context.h 56 torch_xla/csrc/matrix.cpp 126 torch_xla/csrc/matrix.h 12 torch_xla/csrc/nll_loss.cpp 153 torch_xla/csrc/nll_loss.h 12 torch_xla/csrc/nms_op.cpp 214 torch_xla/csrc/nms_op.h 11 torch_xla/csrc/op_by_op_executor.cpp 195 torch_xla/csrc/op_by_op_executor.h 33 torch_xla/csrc/ops/adaptive_avg_pool2d.cpp 42 torch_xla/csrc/ops/adaptive_avg_pool2d.h 20 torch_xla/csrc/ops/adaptive_avg_pool3d.cpp 42 torch_xla/csrc/ops/adaptive_avg_pool3d.h 21 torch_xla/csrc/ops/adaptive_max_pool2d.cpp 43 torch_xla/csrc/ops/adaptive_max_pool2d.h 19 torch_xla/csrc/ops/all.cpp 47 torch_xla/csrc/ops/all.h 23 torch_xla/csrc/ops/all_gather.cpp 59 torch_xla/csrc/ops/all_gather.h 27 torch_xla/csrc/ops/all_reduce.cpp 70 torch_xla/csrc/ops/all_reduce.h 27 torch_xla/csrc/ops/all_to_all.cpp 65 torch_xla/csrc/ops/all_to_all.h 29 torch_xla/csrc/ops/amax.cpp 41 torch_xla/csrc/ops/amax.h 20 torch_xla/csrc/ops/amin.cpp 41 torch_xla/csrc/ops/amin.h 20 torch_xla/csrc/ops/amp_foreach_non_finite_check_and_unscale.cpp 56 torch_xla/csrc/ops/amp_foreach_non_finite_check_and_unscale.h 16 torch_xla/csrc/ops/amp_update_scale.cpp 44 torch_xla/csrc/ops/amp_update_scale.h 20 torch_xla/csrc/ops/any.cpp 47 torch_xla/csrc/ops/any.h 23 torch_xla/csrc/ops/arg_max.cpp 37 torch_xla/csrc/ops/arg_max.h 20 torch_xla/csrc/ops/arg_min.cpp 37 torch_xla/csrc/ops/arg_min.h 20 torch_xla/csrc/ops/arithmetic_ir_ops.cpp 53 torch_xla/csrc/ops/arithmetic_ir_ops.h 10 torch_xla/csrc/ops/as_strided.cpp 84 torch_xla/csrc/ops/as_strided.h 32 torch_xla/csrc/ops/as_strided_view_update.cpp 73 torch_xla/csrc/ops/as_strided_view_update.h 27 torch_xla/csrc/ops/avg_pool_nd.cpp 81 torch_xla/csrc/ops/avg_pool_nd.h 31 torch_xla/csrc/ops/avg_pool_nd_backward.cpp 86 torch_xla/csrc/ops/avg_pool_nd_backward.h 33 torch_xla/csrc/ops/bernoulli.cpp 29 torch_xla/csrc/ops/bernoulli.h 14 torch_xla/csrc/ops/binary_cross_entropy.cpp 64 torch_xla/csrc/ops/binary_cross_entropy.h 22 torch_xla/csrc/ops/binary_cross_entropy_backward.cpp 72 torch_xla/csrc/ops/binary_cross_entropy_backward.h 23 torch_xla/csrc/ops/bitwise_ir_ops.cpp 71 torch_xla/csrc/ops/bitwise_ir_ops.h 11 torch_xla/csrc/ops/cast.cpp 65 torch_xla/csrc/ops/cast.h 26 torch_xla/csrc/ops/cat.cpp 46 torch_xla/csrc/ops/cat.h 19 torch_xla/csrc/ops/cholesky.cpp 28 torch_xla/csrc/ops/cholesky.h 18 torch_xla/csrc/ops/collective_permute.cpp 54 torch_xla/csrc/ops/collective_permute.h 24 torch_xla/csrc/ops/constant.cpp 27 torch_xla/csrc/ops/constant.h 18 torch_xla/csrc/ops/constant_pad_nd.cpp 53 torch_xla/csrc/ops/constant_pad_nd.h 22 torch_xla/csrc/ops/convolution_backward_overrideable.cpp 82 torch_xla/csrc/ops/convolution_backward_overrideable.h 36 torch_xla/csrc/ops/convolution_overrideable.cpp 105 torch_xla/csrc/ops/convolution_overrideable.h 43 torch_xla/csrc/ops/cumprod.cpp 58 torch_xla/csrc/ops/cumprod.h 23 torch_xla/csrc/ops/cumsum.cpp 57 torch_xla/csrc/ops/cumsum.h 23 torch_xla/csrc/ops/device_data.cpp 28 torch_xla/csrc/ops/device_data.h 22 torch_xla/csrc/ops/diagonal.cpp 56 torch_xla/csrc/ops/diagonal.h 26 torch_xla/csrc/ops/diagonal_view_update.cpp 35 torch_xla/csrc/ops/diagonal_view_update.h 23 torch_xla/csrc/ops/discrete_uniform.cpp 26 torch_xla/csrc/ops/discrete_uniform.h 15 torch_xla/csrc/ops/expand.cpp 38 torch_xla/csrc/ops/expand.h 19 torch_xla/csrc/ops/exponential.cpp 28 torch_xla/csrc/ops/exponential.h 14 torch_xla/csrc/ops/flip.cpp 26 torch_xla/csrc/ops/flip.h 19 torch_xla/csrc/ops/gather.cpp 43 torch_xla/csrc/ops/gather.h 18 torch_xla/csrc/ops/generic.cpp 32 torch_xla/csrc/ops/generic.h 26 torch_xla/csrc/ops/generic_slice.cpp 47 torch_xla/csrc/ops/generic_slice.h 24 torch_xla/csrc/ops/get_dimensions_size.cpp 34 torch_xla/csrc/ops/get_dimensions_size.h 19 torch_xla/csrc/ops/hardshrink.cpp 27 torch_xla/csrc/ops/hardshrink.h 19 torch_xla/csrc/ops/hardtanh_backward.cpp 35 torch_xla/csrc/ops/hardtanh_backward.h 22 torch_xla/csrc/ops/index_get.cpp 42 torch_xla/csrc/ops/index_get.h 19 torch_xla/csrc/ops/index_ops.cpp 285 torch_xla/csrc/ops/index_ops.h 33 torch_xla/csrc/ops/index_put.cpp 37 torch_xla/csrc/ops/index_put.h 21 torch_xla/csrc/ops/index_select.cpp 40 torch_xla/csrc/ops/index_select.h 18 torch_xla/csrc/ops/infer_output_shape.cpp 21 torch_xla/csrc/ops/infer_output_shape.h 13 torch_xla/csrc/ops/kth_value.cpp 42 torch_xla/csrc/ops/kth_value.h 22 torch_xla/csrc/ops/l1_loss.cpp 41 torch_xla/csrc/ops/l1_loss.h 20 torch_xla/csrc/ops/l1_loss_backward.cpp 48 torch_xla/csrc/ops/l1_loss_backward.h 21 torch_xla/csrc/ops/leaky_relu.cpp 26 torch_xla/csrc/ops/leaky_relu.h 19 torch_xla/csrc/ops/leaky_relu_backward.cpp 31 torch_xla/csrc/ops/leaky_relu_backward.h 20 torch_xla/csrc/ops/linear_interpolation.cpp 29 torch_xla/csrc/ops/linear_interpolation.h 18 torch_xla/csrc/ops/log_softmax.cpp 48 torch_xla/csrc/ops/log_softmax.h 23 torch_xla/csrc/ops/log_softmax_backward.cpp 32 torch_xla/csrc/ops/log_softmax_backward.h 19 torch_xla/csrc/ops/logsumexp.cpp 50 torch_xla/csrc/ops/logsumexp.h 23 torch_xla/csrc/ops/masked_fill.cpp 36 torch_xla/csrc/ops/masked_fill.h 19 torch_xla/csrc/ops/masked_scatter.cpp 24 torch_xla/csrc/ops/masked_scatter.h 14 torch_xla/csrc/ops/masked_select.cpp 35 torch_xla/csrc/ops/masked_select.h 14 torch_xla/csrc/ops/max_in_dim.cpp 41 torch_xla/csrc/ops/max_in_dim.h 20 torch_xla/csrc/ops/max_pool_nd.cpp 74 torch_xla/csrc/ops/max_pool_nd.h 29 torch_xla/csrc/ops/max_pool_nd_backward.cpp 82 torch_xla/csrc/ops/max_pool_nd_backward.h 30 torch_xla/csrc/ops/max_unpool_nd.cpp 54 torch_xla/csrc/ops/max_unpool_nd.h 19 torch_xla/csrc/ops/max_unpool_nd_backward.cpp 63 torch_xla/csrc/ops/max_unpool_nd_backward.h 20 torch_xla/csrc/ops/mean.cpp 64 torch_xla/csrc/ops/mean.h 27 torch_xla/csrc/ops/min_in_dim.cpp 41 torch_xla/csrc/ops/min_in_dim.h 20 torch_xla/csrc/ops/mse_loss.cpp 43 torch_xla/csrc/ops/mse_loss.h 20 torch_xla/csrc/ops/mse_loss_backward.cpp 51 torch_xla/csrc/ops/mse_loss_backward.h 21 torch_xla/csrc/ops/native_batch_norm_backward.cpp 63 torch_xla/csrc/ops/native_batch_norm_backward.h 22 torch_xla/csrc/ops/native_batch_norm_forward.cpp 84 torch_xla/csrc/ops/native_batch_norm_forward.h 22 torch_xla/csrc/ops/nll_loss.cpp 72 torch_xla/csrc/ops/nll_loss.h 24 torch_xla/csrc/ops/nll_loss2d.cpp 72 torch_xla/csrc/ops/nll_loss2d.h 24 torch_xla/csrc/ops/nll_loss2d_backward.cpp 88 torch_xla/csrc/ops/nll_loss2d_backward.h 25 torch_xla/csrc/ops/nll_loss_backward.cpp 88 torch_xla/csrc/ops/nll_loss_backward.h 25 torch_xla/csrc/ops/nms.cpp 55 torch_xla/csrc/ops/nms.h 19 torch_xla/csrc/ops/nonzero.cpp 33 torch_xla/csrc/ops/nonzero.h 14 torch_xla/csrc/ops/normal.cpp 22 torch_xla/csrc/ops/normal.h 14 torch_xla/csrc/ops/not_supported.cpp 25 torch_xla/csrc/ops/not_supported.h 19 torch_xla/csrc/ops/ops.cpp 884 torch_xla/csrc/ops/ops.h 142 torch_xla/csrc/ops/permute.cpp 47 torch_xla/csrc/ops/permute.h 22 torch_xla/csrc/ops/prod.cpp 70 torch_xla/csrc/ops/prod.h 26 torch_xla/csrc/ops/put.cpp 30 torch_xla/csrc/ops/put.h 19 torch_xla/csrc/ops/qr.cpp 52 torch_xla/csrc/ops/qr.h 18 torch_xla/csrc/ops/reduce_scatter.cpp 68 torch_xla/csrc/ops/reduce_scatter.h 30 torch_xla/csrc/ops/reflection_pad2d.cpp 41 torch_xla/csrc/ops/reflection_pad2d.h 19 torch_xla/csrc/ops/reflection_pad2d_backward.cpp 45 torch_xla/csrc/ops/reflection_pad2d_backward.h 20 torch_xla/csrc/ops/repeat.cpp 41 torch_xla/csrc/ops/repeat.h 19 torch_xla/csrc/ops/replication_pad.cpp 41 torch_xla/csrc/ops/replication_pad.h 19 torch_xla/csrc/ops/replication_pad_backward.cpp 46 torch_xla/csrc/ops/replication_pad_backward.h 20 torch_xla/csrc/ops/resize.cpp 35 torch_xla/csrc/ops/resize.h 18 torch_xla/csrc/ops/rrelu_with_noise.cpp 37 torch_xla/csrc/ops/rrelu_with_noise.h 25 torch_xla/csrc/ops/rrelu_with_noise_backward.cpp 39 torch_xla/csrc/ops/rrelu_with_noise_backward.h 25 torch_xla/csrc/ops/scalar.cpp 96 torch_xla/csrc/ops/scalar.h 24 torch_xla/csrc/ops/scatter.cpp 31 torch_xla/csrc/ops/scatter.h 19 torch_xla/csrc/ops/scatter_add.cpp 34 torch_xla/csrc/ops/scatter_add.h 19 torch_xla/csrc/ops/select.cpp 53 torch_xla/csrc/ops/select.h 30 torch_xla/csrc/ops/sgd_optimizer_step.cpp 54 torch_xla/csrc/ops/sgd_optimizer_step.h 22 torch_xla/csrc/ops/shrink_backward.cpp 30 torch_xla/csrc/ops/shrink_backward.h 20 torch_xla/csrc/ops/softmax.cpp 48 torch_xla/csrc/ops/softmax.h 23 torch_xla/csrc/ops/softmax_backward.cpp 32 torch_xla/csrc/ops/softmax_backward.h 19 torch_xla/csrc/ops/softshrink.cpp 27 torch_xla/csrc/ops/softshrink.h 19 torch_xla/csrc/ops/split.cpp 46 torch_xla/csrc/ops/split.h 22 torch_xla/csrc/ops/squeeze.cpp 46 torch_xla/csrc/ops/squeeze.h 18 torch_xla/csrc/ops/stack.cpp 46 torch_xla/csrc/ops/stack.h 19 torch_xla/csrc/ops/std.cpp 53 torch_xla/csrc/ops/std.h 25 torch_xla/csrc/ops/std_mean.cpp 57 torch_xla/csrc/ops/std_mean.h 25 torch_xla/csrc/ops/sum.cpp 62 torch_xla/csrc/ops/sum.h 26 torch_xla/csrc/ops/svd.cpp 79 torch_xla/csrc/ops/svd.h 20 torch_xla/csrc/ops/symeig.cpp 60 torch_xla/csrc/ops/symeig.h 20 torch_xla/csrc/ops/threshold.cpp 28 torch_xla/csrc/ops/threshold.h 20 torch_xla/csrc/ops/threshold_backward.cpp 29 torch_xla/csrc/ops/threshold_backward.h 19 torch_xla/csrc/ops/topk.cpp 48 torch_xla/csrc/ops/topk.h 27 torch_xla/csrc/ops/triangular_solve.cpp 90 torch_xla/csrc/ops/triangular_solve.h 25 torch_xla/csrc/ops/tril.cpp 26 torch_xla/csrc/ops/tril.h 18 torch_xla/csrc/ops/triu.cpp 26 torch_xla/csrc/ops/triu.h 18 torch_xla/csrc/ops/uniform.cpp 26 torch_xla/csrc/ops/uniform.h 15 torch_xla/csrc/ops/unselect.cpp 38 torch_xla/csrc/ops/unselect.h 25 torch_xla/csrc/ops/unsqueeze.cpp 34 torch_xla/csrc/ops/unsqueeze.h 18 torch_xla/csrc/ops/update_slice.cpp 44 torch_xla/csrc/ops/update_slice.h 22 torch_xla/csrc/ops/upsample_bilinear2d.cpp 39 torch_xla/csrc/ops/upsample_bilinear2d.h 22 torch_xla/csrc/ops/upsample_bilinear2d_backward.cpp 42 torch_xla/csrc/ops/upsample_bilinear2d_backward.h 26 torch_xla/csrc/ops/upsample_nearest2d.cpp 35 torch_xla/csrc/ops/upsample_nearest2d.h 19 torch_xla/csrc/ops/upsample_nearest2d_backward.cpp 39 torch_xla/csrc/ops/upsample_nearest2d_backward.h 23 torch_xla/csrc/ops/user_computation.cpp 45 torch_xla/csrc/ops/user_computation.h 19 torch_xla/csrc/ops/var.cpp 54 torch_xla/csrc/ops/var.h 25 torch_xla/csrc/ops/var_mean.cpp 60 torch_xla/csrc/ops/var_mean.h 25 torch_xla/csrc/ops/view.cpp 42 torch_xla/csrc/ops/view.h 18 torch_xla/csrc/ops/xla_ops.cpp 29 torch_xla/csrc/ops/xla_ops.h 45 torch_xla/csrc/pooling.cpp 621 torch_xla/csrc/pooling.h 54 torch_xla/csrc/python_util.cpp 47 torch_xla/csrc/python_util.h 16 torch_xla/csrc/random.cpp 187 torch_xla/csrc/random.h 10 torch_xla/csrc/reduction.cpp 479 torch_xla/csrc/reduction.h 60 torch_xla/csrc/resize_ops.cpp 86 torch_xla/csrc/resize_ops.h 18 torch_xla/csrc/shape_builder.cpp 40 torch_xla/csrc/shape_builder.h 23 torch_xla/csrc/softmax_builder.cpp 82 torch_xla/csrc/softmax_builder.h 10 torch_xla/csrc/tensor.cpp 1379 torch_xla/csrc/tensor.h 942 torch_xla/csrc/tensor_impl.cpp 131 torch_xla/csrc/tensor_impl.h 34 torch_xla/csrc/tensor_methods.cpp 2517 torch_xla/csrc/tensor_ops.cpp 225 torch_xla/csrc/tensor_ops.h 28 torch_xla/csrc/tensor_util.cpp 950 torch_xla/csrc/tensor_util.h 43 torch_xla/csrc/token_handler.cpp 39 torch_xla/csrc/token_handler.h 12 torch_xla/csrc/torch_util.cpp 37 torch_xla/csrc/torch_util.h 41 torch_xla/csrc/version.h 4 torch_xla/csrc/view.cpp 201 torch_xla/csrc/view.h 111 torch_xla/csrc/xla_lower_util.cpp 789 torch_xla/csrc/xla_lower_util.h 75 torch_xla/csrc/xla_op_builder.cpp 813 torch_xla/csrc/xla_op_builder.h 22