- Sort Score
- Result 10 results
- Languages All
Results 51 - 57 of 57 for 2xi64 (0.13 sec)
-
tensorflow/compiler/mlir/lite/tests/legalize-tf-variables.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config_test.cc
using func::FuncOp; using mlir::ModuleOp; static constexpr char kMlirModuleStr[] = R"( module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 1442 : i32}} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 30 03:31:01 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.td
func.func @main(%arg0: tensor<f32>) { %elem = "tf.Const"() <{value = dense<0.000000e+00> : tensor<f32>}> : () -> tensor<f32> %shape = "tf.Const"() <{value = dense<> : tensor<0xi64>}> : () -> tensor<0xi64> %zero = "tf.Fill"(%shape, %elem) : (tensor<0xi64>, tensor<f32>) -> tensor<f32> tf_device.replicate([%arg0, %zero] as %arg1: tensor<f32>) {n = 2 : i32} { %arg1_id = "tf_device.launch"() <{device = "TPU_REPLICATED_HOST_0"}> ({
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 19.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/order_by_dialect.mlir
%7 = "mhlo.broadcast_in_dim"(%3) <{broadcast_dimensions = dense<3> : tensor<1xi64>}> : (tensor<5xf32>) -> tensor<32x28x28x5xf32> %8 = mhlo.add %6, %7 : tensor<32x28x28x5xf32> %9 = mhlo.maximum %8, %1 : tensor<32x28x28x5xf32> %10 = "mhlo.reshape"(%9) : (tensor<32x28x28x5xf32>) -> tensor<32x3920xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter_test.cc
using ::xla::XlaComputation; using ::xla::XlaOp; static constexpr char kMlirModuleStr[] = R"( module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 1442 : i32}} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
SmallVector<Value> start_indices(rank); for (auto [i, start_index] : llvm::enumerate(op.getStartIndices())) { // Start indices should be casted from tensor<i64> to tensor<1xi64>. auto cast = rewriter.create<TFL::BitcastOp>( op->getLoc(), single_element_type, start_index); int64_t upper_limit_idx = operand_shape[i] - slice_sizes[i]; auto upper_limit_attr =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
// For example, // // %0 = tf.Pack(%input) {axis = 0} // %input : tensor<2x3xf32> // // can be canonicalized to // // %shape = "tf.Const"() {value = dense<[1, 2, 3]> : tensor<3xi64>} // %0 = tf.Reshape(%input, %shape) struct ConvertPackToReshape : public OpRewritePattern<PackOp> { using OpRewritePattern<PackOp>::OpRewritePattern; LogicalResult matchAndRewrite(PackOp pack_op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)