- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for IsSupportedVariableType (0.16 sec)
-
tensorflow/compiler/mlir/lite/utils/variables_utils.cc
.cast<ShapedType>() .getElementType() .cast<TF::TensorFlowTypeWithSubtype>() .GetSubtypes() .back(); } return IsSupportedVariableType(type); } bool IsSupportedVariableType(ShapedType type) { auto element_type = type.getElementType(); // Check complex types. if (auto complex_type = element_type.dyn_cast<mlir::ComplexType>()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 21 19:32:03 UTC 2021 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/variables_utils.h
namespace mlir { namespace TFL { namespace utils { // Returns true if 'op' has type that is supported by native TFLite // variables. bool IsSupportedVariableType(Operation* op); // Returns true if 'type' is supported by native tflite variables. bool IsSupportedVariableType(ShapedType type); } // namespace utils } // namespace TFL } // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 21 19:32:03 UTC 2021 - 1.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_variables.cc
// not. const char kLegalizeTflVariables[] = "tfl._legalize_tfl_variables"; bool HasSupportedElementType(Operation* op) { return utils::IsSupportedVariableType(op); } bool IsSupportedElementType(ShapedType type) { return utils::IsSupportedVariableType(type); } #include "tensorflow/compiler/mlir/lite/transforms/generated_legalize_variables.inc"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0)