Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for 5x7xi1 (0.16 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/fold-broadcast.mlir

      func.return %1 : tensor<5x7xi1>
      // CHECK: %[[V0:.*]] = "tf.NotEqual"(%arg0, %arg1) <{incompatible_shape_error = true}> : (tensor<5x7xf32>, tensor<7xf32>) -> tensor<5x7xi1>
      // CHECK: %[[V0]] : tensor<5x7xi1>
    }
    
    // CHECK-LABEL: @broadcast_both_operand
    func.func @broadcast_both_operand(%arg0: tensor<7xf32>, %arg1: tensor<5x1xf32>) -> tensor<5x7xf32> {
      %cst = arith.constant dense<[5, 7]> : tensor<2xi64>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 6.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/internal/passes/input_lowering_metrics_pass_test.cc

      module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 268 : i32}} {
        func.func @main() -> () {
          %cst0 = "tf.Const"(){ value = dense<0> : tensor<3x5xi1>} : () -> tensor<3x5xi1>
          %0 = "tf.Where"(%cst0) : (tensor<3x5xi1>) -> tensor<?x2xi64>
          func.return
        }
      })";
    
      CellReader<int64_t> dynamism_counter(kDynamismOpCounterStreamzName);
      CellReader<int64_t> dynamism_function_counter(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 23 19:42:09 UTC 2024
    - 4.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/tests/decompose_optionals.mlir

    // -----
    
    // CHECK-LABEL: @if
    func.func @if() {
      // CHECK-NOT: Optional
      %0 = builtin.unrealized_conversion_cast to tensor<5xi1>
      %1 = "tf.OptionalNone"() : () -> tensor<!tf_type.variant<tensor<f32>>>
      %2 = "tf.If"(%0, %1) <{else_branch = @false, is_stateless = false, then_branch = @true}>
          : (tensor<5xi1>, tensor<!tf_type.variant<tensor<f32>>>) -> (tensor<!tf_type.variant<tensor<f32>>>)
      return
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 4.5K bytes
    - Viewed (0)
Back to top