- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 36 for hasAttr (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
<< " but got " << element_ty; }); return false; } // FusedMatMul kernel does not support grad_a/grad_b attrs if ((matmul->hasAttr("grad_a") && mlir::cast<BoolAttr>(matmul->getAttr("grad_a")).getValue()) || (matmul->hasAttr("grad_b") && mlir::cast<BoolAttr>(matmul->getAttr("grad_b")).getValue())) { (void)rewriter.notifyMatchFailure(matmul, [&](Diagnostic &diag) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0) -
fastapi/utils.py
# performance by avoiding unnecessary cloning if cloned_types is None: cloned_types = _CLONED_TYPES_CACHE original_type = field.type_ if is_dataclass(original_type) and hasattr(original_type, "__pydantic_model__"): original_type = original_type.__pydantic_model__ use_type = original_type if lenient_issubclass(original_type, BaseModel):
Registered: Mon Jun 17 08:32:26 UTC 2024 - Last Modified: Thu Apr 18 19:40:57 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
if (mlir::isa<TF::ResourceType>(tensor_type.getElementType())) { return true; } } return false; } bool IsTPUOp(mlir::Operation* op) { return op->hasAttr(TF::kReplicationInfoAttr); } StringAttr GetReplicationAttr(mlir::Operation* op) { return op->getAttrOfType<StringAttr>(TF::kReplicationInfoAttr); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
} } } }); } // Remove `kSkipIslandOutlining` attributes. for (func::FuncOp func_op : getOperation().getOps<func::FuncOp>()) { if (func_op->hasAttr(mlir::TF::kSkipIslandOutlining)) { func_op->removeAttr(mlir::TF::kSkipIslandOutlining); } } } } // namespace std::unique_ptr<OperationPass<ModuleOp>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
fastapi/_compat.py
return ( _annotation_is_complex(annotation) or _annotation_is_complex(origin) or hasattr(origin, "__pydantic_core_schema__") or hasattr(origin, "__get_pydantic_core_schema__") ) def field_annotation_is_scalar(annotation: Any) -> bool: # handle Ellipsis here to make tuple[int, ...] work nicely
Registered: Mon Jun 17 08:32:26 UTC 2024 - Last Modified: Thu Apr 18 19:40:57 UTC 2024 - 22.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc
// version 8 and above. constexpr StringRef kUsesShapePolymorphismAttr = "jax.uses_shape_polymorphism"; bool IsInLiftedFunc(Operation* op) { if (op == nullptr) return false; return op->getParentOfType<func::FuncOp>()->hasAttr(kFusedFunctionAttr); } bool IsInStableHloOpRegion(Operation* op) { if (op == nullptr) return false; auto parent_op = op->getParentOp(); return parent_op != nullptr && stablehlo::IsStablehloOp(parent_op); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
if (op->hasAttr(TF::kReplicationInfoAttr)) { op->setAttr(TF::kReplicationInfoAttr, attr); } } // Replaces the replication region attribute if it already exists. void UpdateReplicationAttr(TF::TPUCompilationResultOp& op, StringAttr attr) { // Special case for getting the replication region for // TPUCompilationResultsOp. if (op->hasAttr(kTpuCompilationStatus)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
// A map to find an attribute from its identifier. llvm::StringMap<Attribute> identifier_to_attr; for (Operation& inner_op : float_func.getBody().front().getOperations()) { if (!inner_op.hasAttr(kAttrMapAttribute)) continue; // Insert quantization related attribute if they exists. Quantization // attributes are generated in the prepare pass so the attr_map doesn't // contain the attribute names.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
if (walked_op->hasAttr(kXlaOutsideCompilationAttr)) { return WalkResult::interrupt(); } return WalkResult::advance(); }) .wasInterrupted(); } // Returns whether `op` or any ancestors of `op` are outside compiled. bool HasOutsideCompilationAncestor(Operation* op) { while (op) { if (op->hasAttr(kXlaOutsideCompilationAttr)) { return true;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/preprocess_op.cc
PatternRewriter& rewriter) const override { const auto f_attr = mlir::dyn_cast<FlatSymbolRefAttr>(op.getFAttr()); // Non-quantizable op if (!op->hasAttr(kQuantTraitAttrName)) return failure(); StringRef function_name = f_attr.getValue(); // TODO(b/228928859): Improve the getter function to match attributes rather // than function name.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K bytes - Viewed (0)