Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of about 10,000 for gather (0.27 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/tensor_array_ops_decomposition.mlir

      %indices = "tf.Const"() {value = dense<[1, 2]> : tensor<2xi32>} : () -> tensor<2xi32>
      %gather = "tf.TensorArrayGatherV3"(%ta#0, %indices, %ta#1) : (tensor<!tf_type.resource>, tensor<2xi32>, tensor<f32>) -> tensor<2x3xf32>
      func.return %gather : tensor<2x3xf32>
    }
    
    // -----
    
    // Test inferring shape from the element_shape attribute of gather.
    
    // CHECK-LABEL: func @main
    func.func @main() -> tensor<*xf32> {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 49K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions.cc

          : quant_options_(quant_options), test_mode_(false) {}
    
      LiftQuantizableSpotsAsFunctionsPass(
          const LiftQuantizableSpotsAsFunctionsPass& other) {
        quant_options_ = other.quant_options_;
        test_mode_ = other.test_mode_;
        op_set_ = other.op_set_;
        initializeForTest();
      }
    
      StringRef getArgument() const final {
        // This is the argument used to refer to the pass in
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 10 04:07:09 UTC 2024
    - 16.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir

    // CHECK: stablehlo.gather
    // CHECK-NOT: tfl.gather_nd
    // CHECK-NOT: tfl.gather
    
    // -----
    
    // Tests that a float `stablehlo.gather` is not converted to `tfl.gather_nd`.
    
    func.func @gather_float(%arg0: tensor<3x4x2x2xf32>, %arg1: tensor<2x3x2xi64>) -> tensor<2x3x2x2xf32> {
      %0 = "stablehlo.gather"(%arg0, %arg1) {
        dimension_numbers = #stablehlo.gather<
          offset_dims = [2, 3],
    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/quantization/tensorflow/python/integration_test/quantize_model_test_base.py

      ) -> Tuple[core.Tensor, core.Tensor]:
        """Creates a basic gather model.
    
        This is intended to be used for TF1 (graph mode) tests.
    
        Args:
          input_type: type of the input index tensor for gather operation.
          use_variable_for_filter: Setting this to `True` makes the filter for the
            gather operation a `tf.Variable`.
    
        Returns:
          in_placeholder: Input tensor placeholder.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 21 08:51:46 UTC 2024
    - 51.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test_base.py

            if self.same_scale_op == 'concatenate':
              ones = array_ops.ones_like(out)
              out = array_ops.concat([out, ones], 0)
            elif self.same_scale_op == 'gather':
              out = array_ops.gather(out, indices=[0], axis=0)
            elif self.same_scale_op == 'max_pool':
              out = nn_ops.max_pool(out, ksize=3, strides=1, padding='SAME')
            elif self.same_scale_op == 'pad':
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 18.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions_drq.cc

      }
    
      LiftQuantizableSpotsAsFunctionsDRQPass(
          const LiftQuantizableSpotsAsFunctionsDRQPass& other) {
        quantization_method_ = other.quantization_method_;
        target_opset_ = other.target_opset_;
        min_num_elements_for_weights_ = other.min_num_elements_for_weights_;
      }
    
      StringRef getArgument() const final {
        // This is the argument used to refer to the pass in
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.5K bytes
    - Viewed (0)
  7. cluster/validate-cluster.sh

      if [[ ${attempt} -gt 0 ]]; then
        sleep 15
      fi
      attempt=$((attempt+1))
    
      # The "kubectl get nodes -o template" exports node information.
      #
      # Echo the output and gather 2 counts:
      #  - Total number of nodes.
      #  - Number of "ready" nodes.
      #
      # Suppress errors from kubectl output because during cluster bootstrapping
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Tue Oct 01 06:35:39 UTC 2019
    - 7.3K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/encapsulate_util.cc

          // Other edges might have `dst` as dst node. Update those edges with
          // `replace_node`.
          for (int j = i + 1, end = data_edges.size(); j < end; j++) {
            if (data_edges[j].dst == dst) {
              data_edges[j].dst = replace_node;
            }
          }
    
          // Other placeholder node might have `dst` as original node. Update
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 15.5K bytes
    - Viewed (0)
  9. tensorflow/cc/framework/fuzzing/op_fuzzing.bzl

    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Nov 07 19:14:57 UTC 2022
    - 4.3K bytes
    - Viewed (0)
  10. src/cmd/go/testdata/script/cgo_undef.txt

    #
    # Package c imports package a, and uses cgo itself.
    # The generation of the _cgo_import.go for package c will fail,
    # because it won't know that it has to link against a/libb.a
    # (because we don't gather the #cgo LDFLAGS from all transitively
    # imported packages).
    #
    # The _cgo_import.go file is only needed for internal linking.
    # When generating _cgo_import.go for package c fails, an ordinary
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Wed Jun 22 20:56:07 UTC 2022
    - 1.6K bytes
    - Viewed (0)
Back to top