- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for hasAttr (0.1 sec)
-
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/lite/transforms/lower_static_tensor_list.cc
}; return llvm::all_of(op->getOperandTypes(), is_not_variant) && llvm::all_of(op->getResultTypes(), is_not_variant); }; auto is_set_item_legal = [](Operation *op) { return op->hasAttr("resize_if_index_out_of_bounds") && op->getAttr("resize_if_index_out_of_bounds") .cast<mlir::BoolAttr>() .getValue(); }; ConversionTarget target(*context);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0)