- Sort Score
- Result 10 results
- Languages All
Results 51 - 54 of 54 for Dtype (0.03 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc
func::FuncOp BuildGELUDecomposition(RankedTensorType type, PatternRewriter& rewriter, Block* insertion_point) { rewriter.setInsertionPointToStart(insertion_point); auto ftype = FunctionType::get(rewriter.getContext(), {type}, {type}); auto name = rewriter.getStringAttr("gelu_decomp");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/convert_type.cc
tflite::TensorType ConvertTypeToTensorType(mlir::Type type) { if (type.isF16()) { return tflite::TensorType_FLOAT16; } else if (type.isBF16()) { return tflite::TensorType_BFLOAT16; } else if (type.isF32()) { return tflite::TensorType_FLOAT32; } else if (type.isF64()) { return tflite::TensorType_FLOAT64; } else if (mlir::isa<mlir::TF::StringType>(type)) { return tflite::TensorType_STRING;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/folders.cc
return success(); } static LogicalResult FoldDivOp(stablehlo::DivOp op, PatternRewriter& rewriter) { auto etype = op.getType().getElementType(); if (etype.isa<FloatType>()) { return FoldDivOpInternal<APFloat>(op, rewriter); } if (etype.isa<IntegerType>()) { return FoldDivOpInternal<APInt>(op, rewriter); } return failure(); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 06:11:55 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.cc
auto attr_key = attr.first; auto attr_value = attr.second; if (attr_key == attrKey && attr_value.value_case() == AttrValue::kType && attr_value.type() == DT_RESOURCE) { return true; break; } } } } } return false; } bool IsNonReplicatedGraph(const Graph& graph,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 12:22:33 UTC 2024 - 8.9K bytes - Viewed (0)