Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 50 of 122 for Dtype (0.04 sec)

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

          OpT op, typename OpT::Adaptor adaptor,
          ConversionPatternRewriter &rewriter) const override {
        Type dtype = op.getElementDtype();
        if (!(dtype.isF16() || dtype.isF32() || dtype.isF64() ||
              dtype.isInteger(1) || dtype.isInteger(8) || dtype.isInteger(16) ||
              dtype.isInteger(32) || dtype.isInteger(64))) {
          const char *error_info =
              "requires element_dtype to be 1-bit/8-bit/16-bit/32-bit/64-bit "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_distributed_test.cc

          "      attr {"
          "        key: 'dtype'"
          "        value {"
          "          type: DT_FLOAT"
          "        }"
          "      }"
          "    }"
          "    node_def {"
          "      name: 'read1'"
          "      op: 'ReadVariableOp'"
          "      input: 'var'"
          "      device: '/job:localhost/replica:0/task:1/device:CPU:0'"
          "      attr {"
          "        key: 'dtype'"
          "        value {"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 15 09:49:45 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc

      ctx.appendDialectRegistry(registry);
      ctx.loadAllAvailableDialects();
    }
    
    Status ConvertDataTypeToTensor(tensorflow::DataType dtype, Builder builder,
                                   Type* type) {
      Status s = tensorflow::ConvertDataType(dtype, builder, type);
      if (s.ok()) *type = UnrankedTensorType::get(*type);
      return s;
    }
    
    class MlirTensor : public TracingTensorHandle {
     public:
      explicit MlirTensor(Value value)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 28.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/utils/convert_tensor_test.cc

    class ConvertTensorTest : public ::testing::Test {
     protected:
      template <typename T>
      void VerifyConversion(std::initializer_list<T> values, DataType dtype,
                            mlir::Type expected_ty) {
        mlir::Builder b(expected_ty.getContext());
        Tensor tensor(dtype, TensorShape({static_cast<int64_t>(values.size())}));
        tensor.flat<T>().setValues(values);
    
        auto value_or = ConvertTensor(tensor, &b);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.4K bytes
    - Viewed (0)
  5. tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc

      tensorflow::AsyncValueTensor* av_tensor =
          tensorflow::AsyncValueTensor::FromTensor(tensor);
      if (av_tensor == nullptr) {
        TF_ASSIGN_OR_RETURN(
            *tensor, MakeTensorFromPjRtBuffer(tensor->dtype(), tensor->shape(),
                                              std::move(buffer)));
      } else {
        av_tensor->SetBuffer(std::move(buffer));
      }
      return absl::OkStatus();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 05:48:24 UTC 2024
    - 3.7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

            VLOG(1) << "[potentially conservative] Op type `" << node.type_string()
                    << "` is stateful but effects not modelled";
          } else {
            // See if any resource type is used.
            bool resource = false;
            std::function<bool(mlir::Type)> record_resource;
            record_resource = [&](mlir::Type type) {
              type.walk([&](mlir::Type t) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc

    }
    
    Status ConvertAttribute(mlir::Type type, bool remove_ref_type,
                            AttrValue* value) {
      DataType dtype;
      TF_RETURN_IF_ERROR(ConvertToDataType(type, &dtype));
      if (tensorflow::IsRefType(dtype)) dtype = tensorflow::RemoveRefType(dtype);
      value->set_type(dtype);
      return absl::OkStatus();
    }
    
    Status ConvertAttribute(const mlir::TypeAttr& type, bool remove_ref_type,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  8. tensorflow/cc/ops/while_loop.cc

      NodeBuilder::NodeOut enter_input(enter_output.node(), enter_output.index());
    
      const int next_output_index = 0;
      DataType dtype = enter_output.node()->output_type(0);
      NodeBuilder::NodeOut next_input(NextIterationName(scope, loop_var_idx),
                                      next_output_index, dtype);
    
      std::vector<NodeBuilder::NodeOut> input_list({enter_input, next_input});
      const string unique_name = scope.GetUniqueNameForOp("Merge");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 01:01:21 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc

        size_t idx = type_and_idx.index();
        auto result_ty = mlir::cast<mlir::RankedTensorType>(type_and_idx.value());
    
        // If the result type isn't static, then the owner of the result may be a
        // cast op from a more specific bounded type to an unbounded dynamic type.
        // Use the bounded type to get the buffer size.
        mlir::RankedTensorType buffer_ty = result_ty;
        if (!buffer_ty.hasStaticShape()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 17:24:39 UTC 2024
    - 45.3K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api_unified_experimental_graph.cc

            operation->SetOpName(absl::StrCat("_input_", inputs_.size()).c_str()));
        TF_RETURN_IF_ERROR(operation->SetAttrType("dtype", dtype));
        if (!shape.unknown_rank()) {
          TF_RETURN_IF_ERROR(operation->SetAttrShape(
              "shape", reinterpret_cast<int64_t*>(shape.dim_sizes().data()),
              shape.dims()));
        }
        int num_outputs = 1;
        std::vector<AbstractTensorHandle*> outputs(num_outputs);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 20:00:09 UTC 2024
    - 15.4K bytes
    - Viewed (0)
Back to top