- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 61 for hasAttr (0.11 sec)
-
src/testing/slogtest/slogtest.go
}, checks: []check{ hasAttr("a", "b"), hasAttr("k", "v"), }, }, { name: "groups", explanation: withSource("a Handler should handle Group attributes"), f: func(l *slog.Logger) { l.Info("msg", "a", "b", slog.Group("G", slog.String("c", "d")), "e", "f") }, checks: []check{ hasAttr("a", "b"), inGroup("G", hasAttr("c", "d")), hasAttr("e", "f"), }, },
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Wed Nov 08 17:55:47 UTC 2023 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/canonicalize_compile_and_replicate_attributes.cc
} // Convert `_XlaMustCompile`. if (op->hasAttr(tensorflow::kMustCompileAttr)) { bool must_compile_attr_val = op->getAttrOfType<BoolAttr>(tensorflow::kMustCompileAttr).getValue(); op->removeAttr(tensorflow::kMustCompileAttr); if (!must_compile_attr_val) { if (op->hasAttr(tensorflow::kCompileDeviceTypeAttr)) { op->emitOpError()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_validate_inputs.cc
} else { shardings_result.push_back(sharding_next); } } } bool IsValidShardingTupleForArity(Operation* op) { if (!op->hasAttr(kXLAShardingAttr) && !op->hasAttr(kShardingAttr)) { return true; } std::string shard_string; if (op->hasAttr(kXLAShardingAttr)) { shard_string = op->getAttrOfType<StringAttr>(kXLAShardingAttr).strref().str(); } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 06:51:01 UTC 2024 - 21.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/move_tpu_compile_to_front.cc
op = op->getParentOp(); } }); } int OutsideCompilationOrdering(Operation* predecessor, Operation* op) { // Actual compilations go first. if (op->hasAttr("_is_compilation")) return 2; // Followed by nested ops that contain compilations. if (op->hasAttr("_wraps_compilation")) return 1; // Followed by everything else. return 0; } void MoveTpuCompileToFrontPass::runOnOperation() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 17 00:26:18 UTC 2023 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replica_id_to_device_ordinal.cc
return (llvm::isa<TF::EnqueueTPUEmbeddingSparseTensorBatchOp, TF::EnqueueTPUEmbeddingRaggedTensorBatchOp, TF::EnqueueTPUEmbeddingArbitraryTensorBatchOp>(op) && op->hasAttr(kDeviceOrdinalAttr) && op->hasAttr(kReplicaIdAttr)); } void ReplicaIDToDeviceOrdinalPass::runOnOperation() { const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); if (!tf_dialect) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/unwrap_xla_call_module_op.cc
SymbolTable& symbol_table) { // Do not inline lifted quantized functions used for fusing patterns. // TODO - b/310539922: Remove reference to TF/TFL utils. if (call_op->hasAttr(kQuantTraitAttrName)) { return; } auto function_name = call_op ->getAttrOfType<FlatSymbolRefAttr>( TF::kStablehloEntryFunctionAttrName)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.cc
namespace tensorflow { using ::mlir::failure; using ::mlir::LogicalResult; using ::mlir::success; namespace { constexpr absl::string_view kPartitionedCall = "TPUPartitionedCall"; LogicalResult HasAttr( const Graph& graph, const FunctionLibraryDefinition* function_library, const std::function<bool(const Graph& graph)>& predicate) { if (predicate(graph)) { return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 12:22:33 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_program_key.cc
preprocess_op->setOperand(0, program); } LogicalResult VerifyAllProgramKeyOperandsReplaced(Operation* module) { WalkResult result = module->walk([&](Operation* op) { if (!op->hasAttr(kMiniBatchSplitsAttr) && !op->hasAttr(kMiniBatchCsrAttr)) return WalkResult::advance(); Operation* defining = op->getOperand(0).getDefiningOp(); if (llvm::dyn_cast_or_null<TF::ConstOp>(defining)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_to_legacy_compile_and_replicate_attributes.cc
LogicalResult ConvertToLegacyAttributes(func::FuncOp func_op) { auto result = func_op->walk([&](mlir::Operation* op) { if (failed(TF::HasValidCompilationAndReplicationAttributes(*op))) return WalkResult::interrupt(); if (op->hasAttr(TF::kReplicationInfoAttr)) { op->setAttr(TF::kTpuReplicateAttr, op->getAttr(TF::kReplicationInfoAttr)); op->removeAttr(TF::kReplicationInfoAttr); op->removeAttr(TF::kCompileDeviceTypeAttr); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/verify_clustering_pass.cc
op->getDialect()->getNamespace().str() + " not in tf functional dialect"; op->emitError() << error; return WalkResult::interrupt(); } if (op->hasAttr(mlir::TF::kXlaOutsideCompilationAttr)) { std::string error = "op has outside compilation attribute _xla_outside_compilation which " "is not allowed after clustering"; op->emitError() << error;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 08 20:33:37 UTC 2023 - 2.6K bytes - Viewed (0)