- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 26 for blockMark (0.13 sec)
-
src/cmd/compile/internal/ssa/check.go
"math" "math/bits" ) // checkFunc checks invariants of f. func checkFunc(f *Func) { blockMark := make([]bool, f.NumBlocks()) valueMark := make([]bool, f.NumValues()) for _, b := range f.Blocks { if blockMark[b.ID] { f.Fatalf("block %s appears twice in %s!", b, f.Name) } blockMark[b.ID] = true if b.Func != f { f.Fatalf("%s.Func=%s, want %s", b, b.Func.Name, f.Name) }
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Tue Apr 09 16:41:23 UTC 2024 - 17.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
cast_input.setType(transform_result_type); auto block_arg = mlir::dyn_cast<mlir::BlockArgument>(cast_input); auto cast_op_input = dyn_cast_or_null<TF::CastOp>(cast_input.getDefiningOp()); while (block_arg || cast_op_input) { if (block_arg) { // Change on device function type/shape. HandleFuncOp(block_arg.getOwner()->getParentOp()); block_arg = nullptr; cast_op_input = nullptr; } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_invariant_op_hoisting.cc
// If ShapeOp operand is replicate tensor block argument, replace with the // associated first replica operand. if (auto block_arg = mlir::dyn_cast<BlockArgument>(input)) { if (block_arg.getOwner() != replicate_block) return; shape_op.setOperand(replicate_op.GetReplicaOperandForBlockArgument( block_arg, /*replica=*/0)); return; } Operation* input_def = input.getDefiningOp();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
<< ") and block argument " << block_arg.getArgNumber() << " (" << block_arg.getType() << ") to have compatible types"; return success(); }; for (auto block_arg : block.getArguments()) { if (block_arg.getArgNumber() < num_replicated_block_args) { for (int32_t i = n * block_arg.getArgNumber(), e = i + n; i < e; ++i) if (failed(verify_operand_types(block_arg, i))) return failure(); } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
} BlockArgument block_arg; if (failed(AppendReplicatedInput(builder, replicate, inputs, val_bcast.getType(), block_arg))) { return failure(); } OpBuilder before_cluster_builder(cluster); IdentityOp assigned_id = before_cluster_builder.create<IdentityOp>( val_bcast.getLoc(), block_arg.getType(), block_arg);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/annotate_parameter_replication.cc
auto operand = SkipIdentityAndReadVariable(entry.value()); auto block_arg = mlir::dyn_cast<BlockArgument>(operand); if (block_arg && block_arg.getOwner() == &replicate.GetBody()) { // Only mirrored args of ReplicateOp can be annotated. if (mirrored_replicate_args.count(block_arg.getArgNumber()) == 0) { continue; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_device_copy_conversion.cc
return true; } if (BlockArgument block_arg = mlir::dyn_cast<BlockArgument>(arg)) { // Skip the folding logic if the block argument is not from the function // arguments. This can happen when the argument is from a while loop. if (block_arg.getParentRegion() != &func_op.getRegion()) { return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc
} seen.insert(v); if (auto blockArg = mlir::dyn_cast<BlockArgument>(v)) { Operation *op = blockArg.getOwner()->getParentOp(); // If we're in the first block, then the argument to that block is the // one we're looking for. if (auto func = cast<func::FuncOp>(op)) { if (blockArg.getOwner()->isEntryBlock()) { *out = blockArg; return success(); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_host_computation_expansion.cc
} // Finds op that created a given value. If the value is a BlockArgument, this // returns the owner of the Block. Operation* GetOpOfValue(Value value) { if (auto block_arg = mlir::dyn_cast<BlockArgument>(value)) return block_arg.getOwner()->getParentOp(); return value.getDefiningOp(); } // TODO(b/158596585): Replace this with a cost model analysis. bool IsTrivialUnaryOperation(Operation* op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
continue; } auto block_arg = mlir::dyn_cast<BlockArgument>(arg); if (!block_arg || block_arg.getOwner() != &replicate.GetBody()) continue; assert(replicate_arg_to_execute_arg.count(block_arg.getArgNumber()) == 0 && "Found duplicate use of a resource in the execute op."); replicate_arg_to_execute_arg[block_arg.getArgNumber()] = index_and_arg.index(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0)