Home
last modified time | relevance | path

Searched refs:binding_index (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_engine_utils.cc76 int binding_index, bool use_implicit_batch, in GetTrtBindingShape() argument
80 dims = cuda_engine->getBindingDimensions(binding_index); in GetTrtBindingShape()
85 dims = execution_context->getBindingDimensions(binding_index); in GetTrtBindingShape()
107 int* binding_index) { in GetTrtBindingIndex() argument
114 *binding_index = cuda_engine->getBindingIndex(tensor_name); in GetTrtBindingIndex()
115 if (*binding_index == -1) { in GetTrtBindingIndex()
130 *binding_index = *binding_index + profile_index * bindings_per_profile; in GetTrtBindingIndex()
145 int binding_index; in SetTrtEngineInputs() local
147 cuda_engine, &binding_index)); in SetTrtEngineInputs()
170 execution_context->setBindingDimensions(binding_index, trt_dims); in SetTrtEngineInputs()
[all …]
Dtrt_engine_utils.h130 int* binding_index);
148 int binding_index, bool use_implicit_batch,
/external/mesa3d/src/mesa/main/
Dglthread_draw.c132 unsigned binding_index = vao->Attrib[i].BufferIndex; in upload_vertices() local
134 if (!(user_buffer_mask & (1 << binding_index))) in upload_vertices()
137 unsigned stride = vao->Attrib[binding_index].Stride; in upload_vertices()
138 unsigned instance_div = vao->Attrib[binding_index].Divisor; in upload_vertices()
163 unsigned binding_index_bit = 1u << binding_index; in upload_vertices()
167 start_offset[binding_index] = offset; in upload_vertices()
168 end_offset[binding_index] = offset + size; in upload_vertices()
170 if (offset < start_offset[binding_index]) in upload_vertices()
171 start_offset[binding_index] = offset; in upload_vertices()
172 if (offset + size > end_offset[binding_index]) in upload_vertices()
[all …]
Dglthread_varray.c217 enable_buffer(struct glthread_vao *vao, unsigned binding_index) in enable_buffer() argument
219 int attrib_count = ++vao->Attrib[binding_index].EnabledAttribCount; in enable_buffer()
222 vao->BufferEnabled |= 1 << binding_index; in enable_buffer()
224 vao->BufferInterleaved |= 1 << binding_index; in enable_buffer()
228 disable_buffer(struct glthread_vao *vao, unsigned binding_index) in disable_buffer() argument
230 int attrib_count = --vao->Attrib[binding_index].EnabledAttribCount; in disable_buffer()
233 vao->BufferEnabled &= ~(1 << binding_index); in disable_buffer()
235 vao->BufferInterleaved &= ~(1 << binding_index); in disable_buffer()
/external/mesa3d/src/amd/vulkan/
Dradv_descriptor_set.h119 unsigned binding_index) in radv_immutable_ycbcr_samplers() argument
126 if (offsets[binding_index] == 0) in radv_immutable_ycbcr_samplers()
128 return (const struct radv_sampler_ycbcr_conversion *)((const char*)set + offsets[binding_index]); in radv_immutable_ycbcr_samplers()
/external/vulkan-validation-layers/layers/
Dobject_tracker_utils.cpp487 … for (uint32_t binding_index = 0; binding_index < pCreateInfo->bindingCount; ++binding_index) { in PreCallValidateCreateDescriptorSetLayout() local
488 const VkDescriptorSetLayoutBinding &binding = pCreateInfo->pBindings[binding_index]; in PreCallValidateCreateDescriptorSetLayout()