- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for IsF32ShapedType (0.15 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/common/utils.cc
bool float_type_observed = false; bool int8_type_observed = false; bool uint8_type_observed = false; for (auto& input : op->getOpOperands()) { auto input_type = input.get().getType(); if (IsF32ShapedType(input_type)) { float_type_observed = true; } else if (IsQI8Type(input_type)) { int8_type_observed = true; } else if (IsQUI8Type(input_type)) { uint8_type_observed = true; } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 05:37:07 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/common/utils.h
// Returns true if 'op' is not TFL Quant / Dequant op. Returns False otherwise // or if 'op' is null. bool NotTFLQuantDequantizeOp(Operation* op); // Returns true if it is a shaped type of f32 elements. inline bool IsF32ShapedType(Type t) { if (auto shaped_type = mlir::dyn_cast_or_null<ShapedType>(t)) { return shaped_type.getElementType().isF32(); } return false; } // Return true when the given element_type is QI8.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
if (a.size() > b.size()) return false; return std::equal(a.rbegin(), a.rend(), b.rbegin()); } // Returns true if it is a shaped type of f32 elements. inline bool IsF32ShapedType(Type t) { if (auto shaped_type = t.dyn_cast_or_null<ShapedType>()) { return shaped_type.getElementType().isF32(); } return false; } // Returns true if it is a shaped type of bf16 elements.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0)