- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for Gpu_Op (0.08 sec)
-
tensorflow/compiler/mlir/tfrt/ir/gpu_ops.td
let description = [{ The TFRT GPU Dialect. }]; let cppNamespace = "::tfrt::gpu"; } class Gpu_Op<string mnemonic, list<Trait> traits = []> : Op<TFRT_GPU_Dialect, mnemonic, traits> { } // TODO(b/260267885): We may add a device argument when we want to support // GPU MIG. def TransferToDeviceOp: Gpu_Op<"transfer_to_device"> { let summary = "Transfer a CPU tensor to device."; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 18 15:01:21 UTC 2024 - 3.3K bytes - Viewed (0) -
src/runtime/defs_darwin_amd64.go
type floatstate64 struct { fpu_reserved [2]int32 fpu_fcw fpcontrol fpu_fsw fpstatus fpu_ftw uint8 fpu_rsrv1 uint8 fpu_fop uint16 fpu_ip uint32 fpu_cs uint16 fpu_rsrv2 uint16 fpu_dp uint32 fpu_ds uint16 fpu_rsrv3 uint16 fpu_mxcsr uint32 fpu_mxcsrmask uint32 fpu_stmm0 regmmst fpu_stmm1 regmmst fpu_stmm2 regmmst
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Fri Apr 12 21:17:22 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
islands_set, wrapped_ops); if (cluster_to_tpu_ops_map.count(cluster_name)) { for (auto tpu_op : cluster_to_tpu_ops_map[cluster_name]) { islands_set.insert(tpu_op); wrapped_ops.insert(&dyn_cast<IslandOp>(*tpu_op).GetBody().front()); } } // Get the sequential order of the candidate islands in the block.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util_test.cc
{{"scale"}, "Cast", {"two"}, {{"SrcT", DT_INT64}, {"DstT", DT_FLOAT}}}, {{"y"}, "Mul", {"x", "scale"}, {{"T", DT_FLOAT}}}, {{"tpu_op"}, "TPUPartitionedCall", {}, {{"Tout", DT_FLOAT}}}, }); tensorflow::set_tf2_execution(true); FunctionDefLibrary flib; *flib.add_function() = fd;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 19:51:50 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
void TensorHandleSilentCopy(bool async, TFE_ContextDevicePlacementPolicy global_policy, TFE_ContextDevicePlacementPolicy thread_policy, bool cpu_op) { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status( TF_NewStatus(), TF_DeleteStatus); TFE_ContextOptions* opts = TFE_NewContextOptions();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0)