Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 17 of 17 for vec (0.26 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc

      for (auto it = func.begin(); it != func.end(); ++it) ++num_ops;
      return num_ops;
    }
    
    std::vector<Value> ResultsAsVector(Operation* op) {
      std::vector<Value> vec;
      vec.reserve(op->getNumResults());
      for (auto res : op->getResults()) vec.push_back(res);
      return vec;
    }
    
    void SetBasicBlockAttributes(OpBuilder& builder, Operation* op) {
      op->setAttr(kDevice, builder.getStringAttr(""));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 92.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc

      {
        llvm::SmallVector<DictionaryAttr, 8> tmp;
        func.getAllResultAttrs(tmp);
    
        for (const auto& res : tmp) {
          result_attrs.push_back(res.getValue().vec());
        }
      }
    
      mlir::Builder builder(func.getOperation());
    
      for (OpOperand& operand : fetch.getOperation()->getOpOperands()) {
        if (mlir::isa<tf_executor::ControlType>(operand.get().getType())) break;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.cc

            mlir::cast<quant::UniformQuantizedPerAxisType>(filter_state.params);
        std::vector<double> new_bias_scales = bias_quantized_type.getScales().vec();
        std::vector<double> new_filter_scales =
            filter_quantized_type.getScales().vec();
    
        bool needs_adjustment = false;
        for (int i = 0; i < bias_quantized_type.getScales().size(); ++i) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 38.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/quantization/lite/quantize_model_test.cc

      return FlatBufferModel::BuildFromFile(model_path.c_str());
    }
    
    template <typename T>
    std::vector<T> GetAsVector(const flatbuffers::Vector<T>* vec) {
      return std::vector<T>(vec->begin(), vec->end());
    }
    
    void VerifyQuantizationScale(
        const QuantizationParameters& float_quant_params,
        const QuantizationParametersT& quantized_quant_params, const int bit_num,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 73.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

      auto edge_padding_low =
          builder_.CreateVector(pad_op.getEdgePaddingLow().vec());
      auto edge_padding_high =
          builder_.CreateVector(pad_op.getEdgePaddingHigh().vec());
      auto interior_padding =
          builder_.CreateVector(pad_op.getInteriorPadding().vec());
    
      auto pad_option = tflite::CreateStablehloPadOptions(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

                                bool is_kernel = false) -> mlir::Value {
          std::vector<int64_t> tensor_shape =
              mlir::cast<ShapedType>(tensor.getType()).getShape().vec();
    
          // Calculate offsets based on depth_idx, channel_idx and tensor_shape
          std::vector<int64_t> start_indices(tensor_shape.size(), 0);
          std::vector<int64_t> limit_indices = tensor_shape;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

    // `tensor<3x2xi32>`.
    static RankedTensorType SubstituteRankedTensorTypeDimSize(
        RankedTensorType input_type, int64_t dim, int64_t dim_size) {
      auto shape = input_type.getShape().vec();
      shape[dim] = dim_size;
      return tensorflow::GetTypeFromTFTensorShape(shape,
                                                  input_type.getElementType());
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
Back to top