Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 22 for dtype (0.14 sec)

  1. tensorflow/c/eager/dlpack_test.cc

      EXPECT_EQ(dltensor_out->device.device_type, dltensor_in->device.device_type);
      EXPECT_EQ(dltensor_out->device.device_id, dltensor_in->device.device_id);
      EXPECT_EQ(dltensor_out->ndim, dltensor_in->ndim);
      EXPECT_EQ(dltensor_out->dtype.code, dltensor_in->dtype.code);
      EXPECT_EQ(dltensor_out->dtype.bits, dltensor_in->dtype.bits);
      EXPECT_EQ(dltensor_out->dtype.lanes, dltensor_in->dtype.lanes);
      for (int i = 0; i < dltensor_in->ndim; ++i) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Jun 30 03:04:46 GMT 2023
    - 4.4K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_unified_experimental.cc

                               TF_Status* s) {
      unwrap(o)->outputs.push_back(unwrap(tensor));
    }
    
    void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type,
                                TF_Status* s) {
      tsl::Set_TF_Status_from_Status(
          s, unwrap(op)->Reset(op_type,
                               /*raw_device_name=*/nullptr));
    }
    
    void TF_AbstractOpSetOpName(TF_AbstractOp* op, const char* const op_name,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  3. tensorflow/c/c_api.cc

        for (const auto& p : *shapes_and_types) {
          auto* out_shape_and_type = handle_data.add_shape_and_type();
          ic->ShapeHandleToProto(p.shape, out_shape_and_type->mutable_shape());
          out_shape_and_type->set_dtype(p.dtype);
          *out_shape_and_type->mutable_type() = p.type;
        }
      }
      string str_data;
      handle_data.SerializeToString(&str_data);
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_experimental.cc

    }
    
    TFE_TensorHandle* TFE_NewTensorHandleFromScalar(TF_DataType data_type,
                                                    void* data, size_t len,
                                                    TF_Status* status) {
      auto dtype = static_cast<tensorflow::DataType>(data_type);
      DCHECK(tensorflow::DataTypeCanUseMemcpy(dtype));
    
      tensorflow::Tensor tensor(dtype, tensorflow::TensorShape({}));
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  5. tensorflow/c/eager/dlpack.cc

      owner->reference.Unref();
      delete owner;
    }
    
    // Converts TF_DATAType to DLPack data type.
    DLDataType GetDlDataType(TF_DataType data_type, TF_Status* status) {
      DLDataType dtype;
      dtype.lanes = 1;
      dtype.bits = TF_DataTypeSize(data_type) * 8;
      switch (data_type) {
        case TF_DataType::TF_BOOL:
          dtype.code = DLDataTypeCode::kDLBool;
          break;
        case TF_DataType::TF_HALF:
        case TF_DataType::TF_FLOAT:
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  6. tensorflow/c/eager/immediate_execution_tensor_handle.cc

      const char* device_name = DeviceName(&s);
      if (!s.ok()) {
        device_name = "<error fetching device name>";
      }
      return absl::StrCat("TensorHandle(", value_string, ", shape=", shape_string,
                          ", dtype=", DataType_Name(DataType()), ", device=\"",
                          device_name, "\")");
    }
    
    Status ImmediateExecutionTensorHandle::SummarizeValue(
        std::string& summary) const {
      Status status;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 2.1K bytes
    - Viewed (0)
  7. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

        }
        if (TFE_TensorHandleDataType(component.get()) != dtype) {
          TF_SetStatus(status, TF_INTERNAL,
                       "Components of a ParallelTensor must all have "
                       "the same dtype");
          return nullptr;
        }
      }
      return std::unique_ptr<ParallelTensor>(
          new ParallelTensor(parallel_device, std::move(components), shape, dtype));
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  8. tensorflow/c/eager/abstract_tensor_handle.cc

        shape_string = "<error computing shape>";
      } else {
        shape_string = shape.DebugString();
      }
      return absl::StrCat("TensorHandle(shape=", shape_string,
                          ", dtype=", DataType_Name(DataType()),
                          ", type=", FullType().DebugString(), ")");
    }
    
    Status AbstractTensorHandle::TensorHandleStatus() const {
      // Tensor handles in current runtime don't carry error info and this method
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 1.4K bytes
    - Viewed (0)
  9. 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 {"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 23.5K bytes
    - Viewed (0)
  10. tensorflow/c/checkpoint_reader.cc

      Status status;
      if (reader_ != nullptr) {
        status = reader_->GetTensor(name, out_tensor);
      } else {
        tensorflow::DataType dtype;
        tensorflow::TensorShape shape;
        status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape);
        if (status.ok()) {
          out_tensor->reset(new Tensor(dtype, shape));
          status = v2_reader_->Lookup(name, out_tensor->get());
          if (!status.ok()) out_tensor->reset();
        }
      }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Aug 25 21:29:12 GMT 2023
    - 5.5K bytes
    - Viewed (0)
Back to top