- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for set_device_ordinal (0.33 sec)
-
tensorflow/compiler/jit/device_compiler_client.cc
if (result.collective_info) { build_options.set_num_replicas(result.collective_info->group_size); } if (options.device_ordinal != -1) { build_options.set_device_ordinal(options.device_ordinal); } else if (default_device_ordinal != -1) { build_options.set_device_ordinal(default_device_ordinal); } build_options.set_result_layout(result.xla_output_shape); build_options.set_device_allocator(options.device_allocator.get());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 26 20:35:26 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/jit/get_compiler_ir.cc
} xla::ExecutableBuildOptions build_options; if (result.collective_info) { build_options.set_num_replicas(result.collective_info->group_size); } build_options.set_device_ordinal( options.device_ordinal != -1 ? options.device_ordinal : local_client->default_device_ordinal()); build_options.set_result_layout(result.xla_output_shape);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tpu_device.cc
} }); auto* const dst_device_to_device_stream_impl = static_cast<tpu::TpuStreamInterface*>(dst_device_to_device_stream); const int dst_device_ordinal = dst_xla_context->stream()->parent()->device_ordinal(); XlaTensor* const xla_input = XlaTensor::FromTensor(input); TF_RET_CHECK(xla_input != nullptr && xla_input->has_shaped_buffer());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
llvm::dyn_cast<mlir::TF::_XlaRecvAtHostOp>(clone)) { recv_at_host.setDeviceOrdinal(core); clone->setOperand(0, core_to_compilation_key[core]); } else if (auto send_from_host = llvm::dyn_cast<mlir::TF::_XlaSendFromHostOp>(clone)) { send_from_host.setDeviceOrdinal(core); clone->setOperand(1, core_to_compilation_key[core]); } else if (auto recv_at_host =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0)