- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for WrapOpInLaunch (0.56 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
builder->getStringAttr(GetRandomStateVariableName()))})); } return state_vars; } // Wraps single op in `tf_device.launch` for explicit device assignment. void WrapOpInLaunch(OpBuilder* builder, Location loc, Operation* op, llvm::StringRef device) { OpBuilder::InsertPoint insert_point = builder->saveInsertionPoint(); auto launch = builder->create<tf_device::LaunchOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
/*compilation_status=*/compilation_status_type, /*program=*/ llvm::SmallVector<Type, 8>(num_cores_per_replica, program_type), compile_op_operands, txt_module, txt_metadata); return tensorflow::WrapOpInLaunch(builder, compile_op.getLoc(), compile_op, compilation_device); } // Assigns explicit devices to replicate op. An aliased device is created per
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
before_cluster_builder.getBoolAttr(true)); std::string device = tensorflow::GetDeviceAliasForHostOfLogicalCore(0); LaunchOp launch = tensorflow::WrapOpInLaunch( &before_cluster_builder, val_bcast.getLoc(), assigned_id, device); Value all_reduce = CreateAllReduce(replicate, inner_builder, launch.getResult(0)); orig_to_new[val_bcast] = all_reduce;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0)