- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for getAttrOfType (0.22 sec)
-
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
FuncOp main_fn = module.lookupSymbol<FuncOp>("main"); if (!main_fn) { int entry_func_count = 0; for (auto fn : module.getOps<FuncOp>()) { auto attrs = fn->getAttrOfType<mlir::DictionaryAttr>("tf.entry_function"); if (attrs && !attrs.empty()) { ++entry_func_count; } } // Verify that module has a least one enrty function.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
TakeDatasetOp, TakeWhileDatasetOp>(op)) return DatasetInput{nullptr, nullptr}; return DatasetInput{op->getAttrOfType<ArrayAttr>("output_shapes"), op->getAttrOfType<ArrayAttr>("output_types")}; } bool ShapeInference::InferShapeForDatasetOpCommon(Operation* op, FuncOp f,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
binary_op->template getAttrOfType<StringAttr>( "fused_activation_function"); if (!binary_op_activation_func || binary_op_activation_func.getValue() != "NONE") return failure(); ShapedType filter_type = filter_cst.getType(); if (llvm::isa<AddOp, SubOp>(binary_op)) { auto padding = fc_op->template getAttrOfType<StringAttr>("padding");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
return success(); } void BatchFunctionOp::eraseArguments(const BitVector& erase_indices) { const StringRef operand_segment_size_attr = getOperandSegmentSizeAttr(); auto operandSegmentSizes = getOperation()->getAttrOfType<DenseI32ArrayAttr>( operand_segment_size_attr); // `operandSegmentSizes` attribute indicates the sizes of the two // variadic operands of `BatchFunctionOp`: `in_tensors` and
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/tensorflow/ir/tf_ops_n_z.cc
// For `NcclAllReduceOp` ops the `device` attribute corresponds to the resource // instance. std::optional<std::string> NcclAllReduceOp::GetResourceInstanceStr() { auto device_attr = (*this)->getAttrOfType<StringAttr>("device"); // Treat missing device attribute like unspecified (= empty string) attribute. // Note that different op instances with the same string (including empty
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
//===----------------------------------------------------------------------===// // Return true if the op has non-empty "minmax" attribute. static inline bool HasValidMinMaxAttribute(Operation* op) { auto minmax = op->getAttrOfType<ArrayAttr>("minmax"); return minmax && minmax.getValue().size() == 2; } namespace { /// This pattern matches and remove a tfl.fake_quant if all the users of this op
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0)