- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 101 for Dtype (0.04 sec)
-
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) -
tensorflow/c/experimental/saved_model/internal/saved_model_api_test.cc
// inputs: { // key : "a" // value: { // name : "serving_default_a:0" // dtype: DT_FLOAT // tensor_shape: { // } // } // } // inputs: { // key : "b" // value: { // name : "serving_default_b:0" // dtype: DT_FLOAT // tensor_shape: { // } // } // } // outputs: {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 08:08:45 UTC 2024 - 21.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
for (auto type : inst.getResultTypes()) if (!mlir::isa<mlir::TensorType, mlir::tf_executor::ControlType, mlir::tf_executor::TokenType>(type)) return errors::InvalidArgument( "Values must be of tensor type, TensorFlow control type, or " "TensorFlow token type. Found ", mlir::debugString(type));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
for (auto type : inst.getResultTypes()) if (!mlir::isa<mlir::TensorType, mlir::tf_executor::ControlType, mlir::tf_executor::TokenType>(type)) return errors::InvalidArgument( "Values must be of tensor type, TensorFlow control type, or " "TensorFlow token type. Found ", mlir::debugString(type));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/hoist_invariant_ops.mlir
// CHECK-NEXT: "tf.AssignVariable"([[v]]#[[handle_idx]], [[v]]#[[const_idx]]) {device = "/CPU:0"} : (tensor<!tf_type.resource<tensor<i32>>>, tensor<i32>) -> () // CHECK-NEXT: [[r:%.*]] = "tf.ReadVariableOp"([[v]]#[[handle_idx]]) {device = "/CPU:0", dtype = i32} : (tensor<!tf_type.resource<tensor<i32>>>) -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 01 23:54:14 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
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({})); std::memcpy(tensorflow::TensorCApi::Buffer(tensor)->data(), data, len); status->status = absl::OkStatus();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
AttrValueMap* values) { AttrValue value; auto& type_list = *value.mutable_list(); for (auto type : types) { DataType dtype; TF_RETURN_IF_ERROR(ConvertScalarTypeToDataType(type, &dtype)); type_list.add_type(dtype); } auto result = values->insert({string(name), value}); assert(result.second && "cannot have multiple attributes with the same name"); (void)result;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_context.cc
shape_determination_fns_.layout_preference_fn( device_tensor->shape(), device_tensor->dtype(), std::nullopt); Status status = [&]() -> Status { TF_ASSIGN_OR_RETURN(xla::Shape shape, shape_determination_fns_.shape_representation_fn( device_tensor->shape(), device_tensor->dtype(), /*fast_mem=*/false, layout_preference));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
tensorflow::CustomDevice* device, tensorflow::DataType dtype, void* data, TFE_CustomDeviceTensorHandleMethods methods) : tensorflow::CustomDeviceTensorHandle(context, device, dtype), data_(data), methods_(methods) {} ~CAPICustomDeviceTensorHandle() override { methods_.deallocator(data_); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0) -
tensorflow/cc/experimental/libtf/value.h
/// @return The unwrapped value. If this `TaggedValue` type does not currently /// contain a value of type `T`, the program terminates via a call to /// `assert`. template <typename T> T& get() { assert(type_ == EnumValueOf<T>::value); return UnionAccess<T>::unsafe_reference(*this); } /// @brief Get the underlying value based on type. /// /// @tparam T The desired return type.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:23:45 UTC 2024 - 20.4K bytes - Viewed (0)