- Sort Score
- Result 10 results
- Languages All
Results 11 - 17 of 17 for isF16 (0.09 sec)
-
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 11:51:44 UTC 2024 - 14.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
Type elem_type = getElementTypeOrSelf(type); // Xla's all_reduce legalizer bitcasts to 32 bits, so only // element types size <= 4 bytes are supported. if (elem_type.isBF16() || elem_type.isF16() || elem_type.isTF32() || elem_type.isF32()) { zero = builder.getFloatAttr(elem_type, 0); } else { return false; } if (auto ranked_type = dyn_cast<RankedTensorType>(type)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize_dynamic_range.cc
ShapedType new_result_type = mlir::dyn_cast<ShapedType>(cast_op.getType()); // Proceeds only if the casting is to float16 if (!new_result_type.getElementType().isF16()) continue; // Cast values std::vector<Eigen::half> new_values; DenseFPElementsAttr value_attr = mlir::cast<DenseFPElementsAttr>(op.getValue());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
// is up to ~2x faster. const bool is_f16 = input_ty.getElementType().isF16(); if (is_f16 && CanUseTensorCores(devices)) return "NHWC"; // For f32/f16 data type decision depends on the filter size in spatial // dimensions, for other data types we keep current data format. if (!input_ty.getElementType().isF32() && !input_ty.getElementType().isF16()) return getDataFormat();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
// initialized to 0. LogicalResult matchAndRewrite( 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 =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
// accumulation over the given input type. Type GetSumAccumulationType(Type input_type) { MLIRContext *ctx = input_type.getContext(); if (input_type.isBF16() || input_type.isF16()) return FloatType::getF32(ctx); if (input_type.isSignlessInteger(8) || input_type.isSignlessInteger(16)) return IntegerType::get(ctx, 32); return input_type; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
"'isSigned' can only be set for 8/16-bits integer type"); } } if (type.isF32()) { return tflite::TensorType_FLOAT32; } else if (type.isF16()) { return tflite::TensorType_FLOAT16; } else if (type.isBF16()) { return tflite::TensorType_BFLOAT16; } else if (type.isF64()) { return tflite::TensorType_FLOAT64;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)