Home
last modified time | relevance | path

Searched refs:executable_idx (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc1629 bool PjRtStreamExecutorExecutable::MustDonateParameter(int executable_idx, in MustDonateParameter() argument
1631 return parameters_that_must_be_donated_[executable_idx].contains(parameter); in MustDonateParameter()
1683 int executable_idx, const RunId& run_id, const ExecuteOptions& options, in EnqueueExecution() argument
1708 bool must_donate = MustDonateParameter(executable_idx, i); in EnqueueExecution()
1772 executables_[executable_idx]->RunAsync(std::move(execution_inputs), in EnqueueExecution()
1801 [references{std::make_tuple(executables_[executable_idx], in EnqueueExecution()
1814 std::make_tuple(executables_[executable_idx], compute_reservation, in EnqueueExecution()
1885 int executable_idx = executables_.size() > 1 ? partition : 0; in ExecuteHelper() local
1890 argument_handles, replica, partition, executable_idx, run_id, options, in ExecuteHelper()
Dpjrt_stream_executor_client.h679 virtual bool MustDonateParameter(int executable_idx, int parameter) const;
690 int partition, int executable_idx, const RunId& run_id,
/external/mesa3d/src/amd/vulkan/
Dradv_pipeline.c5597 for (unsigned i = 0, executable_idx = 0; in radv_GetPipelineExecutablePropertiesKHR() local
5598 i < MESA_SHADER_STAGES && executable_idx < count; ++i) { in radv_GetPipelineExecutablePropertiesKHR()
5601 pProperties[executable_idx].stages = mesa_to_vk_shader_stage(i); in radv_GetPipelineExecutablePropertiesKHR()
5611 pProperties[executable_idx].stages |= VK_SHADER_STAGE_VERTEX_BIT; in radv_GetPipelineExecutablePropertiesKHR()
5625 pProperties[executable_idx].stages |= VK_SHADER_STAGE_TESSELLATION_EVALUATION_BIT; in radv_GetPipelineExecutablePropertiesKHR()
5629 pProperties[executable_idx].stages |= VK_SHADER_STAGE_VERTEX_BIT; in radv_GetPipelineExecutablePropertiesKHR()
5647 pProperties[executable_idx].subgroupSize = pipeline->shaders[i]->info.wave_size; in radv_GetPipelineExecutablePropertiesKHR()
5648 desc_copy(pProperties[executable_idx].name, name); in radv_GetPipelineExecutablePropertiesKHR()
5649 desc_copy(pProperties[executable_idx].description, description); in radv_GetPipelineExecutablePropertiesKHR()
5651 ++executable_idx; in radv_GetPipelineExecutablePropertiesKHR()
[all …]