- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 122 for Dtype (0.1 sec)
-
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/mlir/tensorflow/transforms/host_runtime/tpu_metadata_utils_test.cc
ASSERT_TRUE(tsl::protobuf::TextFormat::ParseFromString( R"pb( args { dtype: DT_FLOAT shape { unknown_rank: true } kind: PARAMETER sharding { type: OTHER tile_assignment_dimensions: 2 tile_assignment_dimensions: 1 tile_assignment_devices: 0
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 12 04:22:33 UTC 2023 - 6.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) -
tensorflow/cc/experimental/libtf/function.cc
} Status VerifySupportedSignature(TaggedValue signature) { if (signature.type() == TaggedValue::Type::TENSOR_SPEC) { return ::tensorflow::OkStatus(); } if (signature.type() == TaggedValue::Type::TUPLE) { for (const auto& t : signature.tuple()) { if (t.type() != TaggedValue::Type::TENSOR_SPEC) { break; } } return ::tensorflow::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 04 19:49:06 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/legalize_tf_quant_test.cc
// Set up an arg per arg_shape with the specified type. for (int i = 0; i < arg_shapes.size(); ++i) { auto metadata_arg = metadata_proto.add_args(); metadata_arg->set_kind( tensorflow::tpu::TPUCompileMetadataProto::Arg::PARAMETER); metadata_arg->set_dtype(dtype); } // Set up one dummy retval. metadata_proto.add_retvals();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 18:43:55 UTC 2024 - 7.2K bytes - Viewed (0)