Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for window_dimensions (0.4 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

              << x_name
              << "\nNumber of window dimensions: " << window_dimensions.size()
              << "\nNumber of " << x_name << ": " << x << "\n";
          return false;
        }
      };
    
      if (!(verify_size(window_dimensions.size(), "window_dimensions") &&
            verify_size(window_strides.size(), "window strides") &&
            verify_size(padding.size(), "padding entries") &&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

        for (int64_t i = 0, e = window_dimensions.size(); i < e; ++i) {
          int64_t window_dimension = window_dimensions.getValues<int64_t>()[i];
          if (window_dimension == 1) continue;
          // Cumulative axis already set.
          if (cumulative_axis != -1) return failure();
          // Potential cumulative axis is not the right size.
          if (window_dimension != operand_type.getShape()[i]) return failure();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir

        %1 = stablehlo.maximum %arg2, %arg3 : tensor<!quant.uniform<i8:f32, 3.000000e-01:-5>>
        stablehlo.return %1 : tensor<!quant.uniform<i8:f32, 3.000000e-01:-5>>
      }) {window_dimensions = array<i64: 1, 3, 4, 1>, window_strides = array<i64: 1, 2, 3, 1>} : (tensor<2x9x10x3x!quant.uniform<i8:f32, 3.000000e-01:-5>>, tensor<!quant.uniform<i8:f32, 3.000000e-01:-5>>) -> tensor<2x4x3x3x!quant.uniform<i8:f32, 3.000000e-01:-5>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 17:10:32 UTC 2024
    - 106.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

      uint32_t opcode_index =
          GetOpcodeIndex(op_name, tflite::BuiltinOperator_STABLEHLO_REDUCE_WINDOW);
    
      auto wd = reduce_window_op.getWindowDimensions();
      auto window_dimensions =
          builder_.CreateVector(std::vector<int64_t>(wd.begin(), wd.end()));
      auto window_strides = builder_.CreateVector(
          mlir::GetOptionalVector<int64_t>(reduce_window_op.getWindowStrides()));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

                                      << size << " vs. " << input_rank << ")";
            }
          }
        }
        return success();
      };
    
      if (check(op.getWindowDimensions(), "window_dimensions").failed())
        return failure();
    
      if (check(op.getWindowStrides(), "window_strides").failed()) return failure();
    
      if (check(op.getBaseDilations(), "base_dilations").failed()) return failure();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

        if (op.getBaseDilations().has_value() ||
            op.getWindowDilations().has_value() || op.getPadding().has_value()) {
          return failure();
        }
    
        // Window_dimensions and window_strides should have batch and channel
        // dimension of 1 as they cannot be specified in tfl.max_pool_2d.
        ArrayRef<int64_t> window_dims = op.getWindowDimensions();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir

      %cst_1 = "tf.Const"() {value = dense<[2, 3, 1]> : tensor<3xi32>} : () -> tensor<3xi32>
      // expected-error @+1 {{'tf.XlaSelectAndScatter' op expects the size of window_dimensionsto be equal to the input rank (3 vs. 4)}}
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 23 14:40:35 UTC 2023
    - 236.4K bytes
    - Viewed (0)
Back to top