Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for shape_b (0.21 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

      auto element_type = x.getElementType();
      auto shape_x = x.getShape();
      auto shape_y = y.getShape();
    
      if (shape_x.size() == shape_y.size()) {
        llvm::SmallVector<int64_t, 4> out_shape(shape_x.size());
        for (int i = 0; i < shape_x.size(); i++) {
          auto x_val = shape_x[i];
          auto y_val = shape_y[i];
          out_shape[i] = std::max(x_val, y_val);
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tests/optimize.mlir

      %begin0 = arith.constant dense<0> : tensor<4xi64>
      %shape0 = arith.constant dense<[2,3,4,4]> : tensor<4xi64>
      %begin1 = arith.constant dense<1> : tensor<4xi64>
      %shape1 = arith.constant dense<[1,2,3,4]> : tensor<4xi64>
      %0 = "tfl.slice"(%arg0, %begin0, %shape0) : (tensor<2x3x4x5xf32>, tensor<4xi64>, tensor<4xi64>) -> tensor<2x3x4x4xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 20:31:41 UTC 2024
    - 284.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

      // CHECK-NEXT: %[[SHAPE0:.*]] = shape.shape_of %arg0 : tensor<?xi1> -> tensor<1xindex>
      // CHECK-NEXT: %[[SHAPE1:.*]] = shape.shape_of %arg1 : tensor<?x?x8xi32> -> tensor<3xindex>
      // CHECK-NEXT: %[[SHAPE2:.*]] = shape.shape_of %arg2 : tensor<?x?x8xi32> -> tensor<3xindex>
      // CHECK-NEXT: %[[SHAPEEQ1:.*]] = shape.cstr_eq %[[SHAPE1]], %[[SHAPE2]] : tensor<3xindex>, tensor<3xindex>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
Back to top