Search Options

Results per page
Sort
Preferred Languages
Advance

Results 81 - 90 of 593 for shake (0.04 sec)

  1. tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc

    }
    
    float CalculateBlockSparsity(const ElementsAttr& attr, const ShapedType& type,
                                 const std::vector<int>& block_size) {
      float sparsity = 0;
      std::vector<int> shape(2);
      shape[0] = type.getDimSize(0);
      shape[1] = type.getDimSize(1);
    
      std::vector<int> traversal_order = {};
      std::vector<TfLiteDimensionType> format = {};
      std::vector<int> b_size = {};
      std::vector<int> b_map = {};
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.h

    // certain downstream optimization passes (e.g. XLA).
    inline constexpr std::array<int64_t, 4> kOihwToHwioPermutation = {2, 3, 1, 0};
    
    // Returns true if the value has static shape.
    bool HasStaticShape(Value value);
    
    // Returns true if the value has static shape at given dims.
    bool HasStaticShapeAtDims(Value value, ArrayRef<int> dims);
    
    // Whether `value` has known rank of `rank`. Returns false when it is not a
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/aot/compile.cc

        return errors::Unknown("Couldn't get XLA program shape: ",
                               pshape_or.status().message());
      }
      compile_result->program_shape = pshape_or.value()->ToProto();
      xla::ProgramShapeProto* pshape = &compile_result->program_shape;
    
      // AotXlaComputationInstance::argument_layouts is a vector of Shape
      // pointers. Accumulate the Shape objects themselves in a separate vector
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 08:28:57 UTC 2024
    - 11.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc

        std::unique_ptr<Graph> g(new Graph(&fld));
    
        tensorflow::TensorProto tensor_proto;
        tensor_proto.set_dtype(tensorflow::DT_INT32);
        tensorflow::TensorShapeProto shape;
        shape.add_dim()->set_size(2);
        *tensor_proto.mutable_tensor_shape() = shape;
        for (int i = 0; i < 2; ++i) {
          tensor_proto.add_int_val(1);
        }
        NodeDef const_def;
        TF_CHECK_OK(NodeDefBuilder("const", "Const")
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 41K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/tests/end2end/fake_quant_per_channel_4bit.pbtxt

    node {
      name: "input"
      op: "Placeholder"
      attr {
        key: "dtype"
        value {
          type: DT_FLOAT
        }
      }
      attr {
        key: "shape"
        value {
          shape {
            dim {
              size: 1
            }
            dim {
              size: 1
            }
            dim {
              size: 1
            }
            dim {
              size: 256
            }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 18.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tfr/tests/canonicalize.mlir

    func.func @get_real_shape(%arg0: tensor<1x2xf32>) -> tensor<2xindex> {
      %0 = "tfr.cast"(%arg0) : (tensor<1x2xf32>) -> !tfr.tensor
      %1 = tfr.get_shape %0 -> !shape.shape
      %2 = shape.to_extent_tensor %1 : !shape.shape -> tensor<2xindex>
      func.return %2 : tensor<2xindex>
    
    // CHECK-NEXT: %[[e:.*]] = shape.const_shape [1, 2] : tensor<2xindex>
    // CHECK-NEXT: return %[[e]] : tensor<2xindex>
    }
    
    // CHECK-LABEL: equal
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 11.1K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_patterns.td

    // ReluGrad(gradients, features) = gradients * (features > 0)
    // The condition that $gradients and $features need to have the same shape is
    // implicitly enforced: $zero is created to have the same shape as $features,
    // MHLO_SelectOp enforces that $gradients and $zero have the same shape.
    def : Pat<(TF_ReluGradOp AnyTensor:$gradients, AnyTensor:$features),
              (MHLO_SelectOp
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 34.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc

      for (const std::optional<llvm::ArrayRef<int64_t>>& shape : shapes) {
        TensorShapeProto& tshape = *shape_list.add_shape();
        if (shape.has_value()) {
          for (int64_t dim : *shape) {
            tshape.add_dim()->set_size(mlir::ShapedType::isDynamic(dim) ? -1 : dim);
          }
        } else {
          tshape.set_unknown_rank(true);
        }
      }
    
      // If shape is already set, override it. This can happen if we import
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 11.1K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tfrt/tests/analysis/cost_analysis.mlir

        // expected-remark@+1 {{Cost: 19}}
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Aug 14 15:35:49 UTC 2023
    - 12.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/unroll_batch_matmul.cc

        Value value, ArrayRef<int64_t> shape, Type element_type, Location loc,
        PatternRewriter& rewriter) {
      int64_t shape_rank = shape.size();
      auto shape_spec_type =
          RankedTensorType::get({shape_rank}, rewriter.getIntegerType(64));
      Type resultType = RankedTensorType::get(shape, element_type);
      auto constant_attr = DenseElementsAttr::get(shape_spec_type, shape);
      auto shape_tensor =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.6K bytes
    - Viewed (0)
Back to top