- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for broadcast_sizes (0.35 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc
namespace mlir { namespace odml { static bool isDenseI64Array(llvm::StringRef op_name, llvm::StringRef field_name) { if (op_name == "stablehlo.broadcast" && field_name == "broadcast_sizes") return true; if (op_name == "stablehlo.broadcast_in_dim" && field_name == "broadcast_dimensions") return true; if ((op_name == "stablehlo.convolution" ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 24 06:08:43 UTC 2024 - 10.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// CHECK-DAG: %[[V5:.*]] = "mhlo.broadcast"(%[[V4]]) <{broadcast_sizes = dense<[1, 22, 128]> : tensor<3xi64>}> : (tensor<i32>) -> tensor<1x22x128xi32> // CHECK-DAG: %[[V6:.*]] = mhlo.constant dense<false> : tensor<i1> // CHECK-DAG: %[[V7:.*]] = "mhlo.broadcast"(%[[V6]]) <{broadcast_sizes = dense<[1, 22, 128]> : tensor<3xi64>}> : (tensor<i1>) -> tensor<1x22x128xi1>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0)