- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for isF16 (0.04 sec)
-
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/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)