- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for deim (0.07 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
// the same because all strides are 1). low_padding[dim] = conv_grad_spatial_dim.pad_before - orig_padding[dim].first; high_padding[dim] = conv_grad_spatial_dim.pad_after - orig_padding[dim].second; // Update `out_grad_shape` to result shape of following `PadOp`. out_grad_shape[dim] = low_padding[dim] + high_padding[dim] +
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir
// CHECK: return %[[RES]] func.func @torch_index_select(%arg0: tensor<2x1xf32>, %arg1: tensor<2xi32>) -> tensor<2x1xf32> { %0 = "mhlo.torch_index_select"(%arg0, %arg1) { batch_dims = 0 : i64, dim = 0 : i64 } : (tensor<2x1xf32>, tensor<2xi32>) -> tensor<2x1xf32> func.return %0 : tensor<2x1xf32> } // CHECK-LABEL: func @lowered_cumsum(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 07:26:59 UTC 2024 - 340.2K bytes - Viewed (0)