- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 107 for Dtype (0.04 sec)
-
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/compiler/jit/get_compiler_ir.cc
const Tensor* value = variable.var()->tensor(); arg.type = value->dtype(); arg.shape = value->shape(); arg.initialized = true; } else { // Instead of embedding constant into HLO, // we handle tf.constant as parameter to reduce size. arg.kind = XlaCompiler::Argument::kParameter; arg.type = input->dtype(); arg.shape = input->shape(); } args.push_back(arg);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K 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/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;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/utils/tf_type_utils.cc
namespace mlir::quant::tensorflow { bool IsTFQintType(const Type type) { return mlir::isa<TF::Qint8Type, TF::Qint16Type, TF::Qint32Type, TF::Quint8Type, TF::Quint16Type>(type); } Type GetIntTypeFromTFQint(const Type type) { return TypeSwitch<Type, Type>(type) .Case<TF::Qint8Type>( [&type](Type) { return IntegerType::get(type.getContext(), 8); }) .Case<TF::Qint16Type>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.1K 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/mark_for_compilation_pass_test.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/tf_tfl_flatbuffer_helpers.cc
"'h_scale' type: 'float'} attr : { name: 'max_classes_per_detection' " "type: 'int'} attr : { name: 'max_detections' type: 'int'} attr : { " "name: 'nms_iou_threshold' type: 'float'} attr : { name: " "'nms_score_threshold' type: 'float'} attr : { name: 'num_classes' type: " "'int'} attr : { name: 'w_scale' type: 'float'} attr : { name: 'x_scale' "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:39:37 UTC 2024 - 17.3K bytes - Viewed (0) -
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc
std::vector<int64_t> size_as_vector; }; std::vector<int64_t> IntTensorAsVector(const Tensor& t) { DCHECK(t.dtype() == DT_INT32 || t.dtype() == DT_INT64); std::vector<int64_t> result; result.reserve(t.NumElements()); for (int i = 0; i < t.NumElements(); i++) { int64_t element = t.dtype() == DT_INT32 ? static_cast<int64_t>(t.flat<int32>()(i)) : t.flat<int64_t>()(i);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 13.8K 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)