Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for ConvertToTensorProto (0.2 sec)

  1. tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc

      value->set_f(attr.getValueAsDouble());
      return absl::OkStatus();
    }
    
    Status ConvertAttribute(const mlir::ElementsAttr& attr, AttrValue* value) {
      return ConvertToTensorProto(attr, value->mutable_tensor());
    }
    
    Status ConvertAttribute(const mlir::TF::PlaceholderAttr& attr,
                            AttrValue* value) {
      value->set_placeholder(attr.getValue().str());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.cc

      } else {
        output->reserve(attr.getNumElements());
        for (const T value : attr.getValues<T>())
          output->push_back(Eigen::numext::bit_cast<uint8_t>(value));
      }
    }
    
    Status ConvertToTensorProto(const ElementsAttr attr, TensorProto* output) {
      auto type = attr.getShapedType();
      auto shape = type.getShape();
      DataType output_dtype;
      TF_RETURN_IF_ERROR(ConvertToDataType(type, &output_dtype));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 20.5K bytes
    - Viewed (0)
Back to top