Searched refs:dispatch_index (Results 1 – 6 of 6) sorted by relevance
| /external/mesa3d/src/amd/vulkan/ |
| D | radv_rra.h | 120 uint32_t dispatch_index; member
141 uint32_t dispatch_index : 16; member
|
| D | radv_rra.c | 1466 if (src->dispatch_index != ray_history_index) { in radv_rra_dump_trace()
1467 ray_history_index = src->dispatch_index; in radv_rra_dump_trace()
|
| D | radv_cmd_buffer.c | 12264 …uint32_t dispatch_index = util_dynarray_num_elements(&cmd_buffer->ray_history, struct radv_rra_ray… in radv_trace_trace_rays() local
12276 … device->rra_trace.ray_history_addr + offsetof(struct radv_ray_history_header, dispatch_index), in radv_trace_trace_rays()
12277 &dispatch_index, sizeof(dispatch_index)); in radv_trace_trace_rays()
|
| /external/pytorch/aten/src/ATen/ |
| D | TensorIndexing.h | 555 inline Tensor dispatch_index( in dispatch_index() function
638 return dispatch_index(
662 return dispatch_index(sliced, std::move(tensorIndices));
|
| /external/pytorch/torch/csrc/autograd/ |
| D | python_variable_indexing.cpp | 406 return at::indexing::dispatch_index(sliced, std::move(variableIndices)); in THPVariable_getitem()
|
| /external/mesa3d/src/amd/vulkan/nir/ |
| D | radv_nir_rt_shader.c | 935 … nir_imm_int(b, offsetof(struct radv_ray_history_header, dispatch_index)), .align_mul = 4); in radv_build_end_trace_token()
936 …nir_def *dispatch_index = nir_iadd(b, nir_channel(b, dispatch_indices, 0), nir_channel(b, dispatch… in radv_build_end_trace_token() local
938 dispatch_and_flags = nir_ior(b, dispatch_and_flags, dispatch_index); in radv_build_end_trace_token()
|