| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_engine_utils.cc | 62 int binding_index, bool use_implicit_batch, in GetTrtBindingShape() argument
68 ? cuda_engine->getBindingDimensions(binding_index) in GetTrtBindingShape()
69 : execution_context->getBindingDimensions(binding_index); in GetTrtBindingShape()
84 std::vector<void*>& buffers, int binding_index) { in SetupBindings() argument
87 const auto dtype = cuda_engine->getBindingDataType(binding_index); in SetupBindings()
91 buffers[binding_index] = const_cast<float*>(tensor.flat<float>().data()); in SetupBindings()
94 buffers[binding_index] = in SetupBindings()
100 buffers[binding_index] = const_cast<int32*>(tensor.flat<int32>().data()); in SetupBindings()
104 buffers[binding_index] = const_cast<bool*>(tensor.flat<bool>().data()); in SetupBindings()
137 int binding_index; in SetTrtEngineInputs() local
[all …]
|
| D | trt_shape_optimization_profiles.cc | 423 int binding_index; in SetShapeTensorMask() local
424 Status status = GetTrtBindingIndex(i, 0, engine, &binding_index); in SetShapeTensorMask()
428 is_shape_tensor_[i] = engine->isShapeBinding(binding_index); in SetShapeTensorMask()
519 int input_index, int binding_index, nvinfer1::ICudaEngine* cuda_engine, in SetInputShapeBinding() argument
524 if (cuda_engine->isShapeBinding(binding_index)) { in SetInputShapeBinding()
529 VLOG(2) << "Setting input shape binding for idx " << binding_index in SetInputShapeBinding()
533 binding_index, actual_shape_values_.at(input_index).d); in SetInputShapeBinding()
536 binding_index); in SetInputShapeBinding()
|
| D | trt_engine_utils.h | 59 int binding_index, bool use_implicit_batch,
|
| D | trt_shape_optimization_profiles.h | 243 Status SetInputShapeBinding(int input_index, int binding_index,
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/common/ |
| D | utils.cc | 62 int* binding_index) { in GetTrtBindingIndex() argument
71 *binding_index = cuda_engine->getBindingIndex(tensor_name); in GetTrtBindingIndex()
72 if (*binding_index == -1) { in GetTrtBindingIndex()
82 *binding_index = *binding_index + profile_index * bindings_per_profile; in GetTrtBindingIndex()
88 int* binding_index) { in GetTrtBindingIndex() argument
92 binding_index); in GetTrtBindingIndex()
|
| D | utils.h | 115 int* binding_index);
122 int* binding_index);
|
| /external/mesa3d/src/amd/vulkan/ |
| D | radv_descriptor_set.h | 207 radv_immutable_ycbcr_samplers(const struct radv_descriptor_set_layout *set, unsigned binding_index) in radv_immutable_ycbcr_samplers() argument
214 if (offsets[binding_index] == 0) in radv_immutable_ycbcr_samplers()
216 return (const struct vk_ycbcr_conversion_state *)((const char *)set + offsets[binding_index]); in radv_immutable_ycbcr_samplers()
|
| /external/mesa3d/src/mesa/main/ |
| D | glthread_varray.c | 239 enable_buffer(struct glthread_vao *vao, unsigned binding_index) in enable_buffer() argument
241 int attrib_count = ++vao->Attrib[binding_index].EnabledAttribCount; in enable_buffer()
244 vao->BufferEnabled |= 1 << binding_index; in enable_buffer()
246 vao->BufferInterleaved |= 1 << binding_index; in enable_buffer()
250 disable_buffer(struct glthread_vao *vao, unsigned binding_index) in disable_buffer() argument
252 int attrib_count = --vao->Attrib[binding_index].EnabledAttribCount; in disable_buffer()
255 vao->BufferEnabled &= ~(1 << binding_index); in disable_buffer()
257 vao->BufferInterleaved &= ~(1 << binding_index); in disable_buffer()
|
| D | glthread_draw.c | 140 unsigned binding_index = vao->Attrib[i].BufferIndex; in upload_vertices() local
142 if (!(user_buffer_mask & (1 << binding_index))) in upload_vertices()
145 unsigned stride = vao->Attrib[binding_index].Stride; in upload_vertices()
146 unsigned instance_div = vao->Attrib[binding_index].Divisor; in upload_vertices()
171 unsigned binding_index_bit = 1u << binding_index; in upload_vertices()
175 start_offset[binding_index] = offset; in upload_vertices()
176 end_offset[binding_index] = offset + size; in upload_vertices()
178 if (offset < start_offset[binding_index]) in upload_vertices()
179 start_offset[binding_index] = offset; in upload_vertices()
180 if (offset + size > end_offset[binding_index]) in upload_vertices()
[all …]
|
| /external/mesa3d/src/imagination/vulkan/pds/ |
| D | pvr_pipeline_pds.c | 132 DMA->binding_index); in pvr_write_pds_const_map_entry_vertex_attribute_address()
144 robust_attribute_entry->binding_index = DMA->binding_index; in pvr_write_pds_const_map_entry_vertex_attribute_address()
161 attribute_entry->binding_index = DMA->binding_index; in pvr_write_pds_const_map_entry_vertex_attribute_address()
1095 obb_buffer_size->binding_index = vertex_dma->binding_index; in pvr_pds_generate_vertex_primary_program()
1215 max_index_entry->binding_index = vertex_dma->binding_index; in pvr_pds_generate_vertex_primary_program()
|
| D | pvr_pds.h | 971 uint8_t binding_index; member
1100 uint8_t binding_index; member
1110 uint8_t binding_index; member
1120 uint8_t binding_index; member
1146 uint8_t binding_index; member
|
| /external/mesa3d/src/gallium/auxiliary/gallivm/ |
| D | lp_bld_jit_types.c | 71 LLVMValueRef binding_index = LLVMBuildExtractValue(builder, index, 1, ""); in lp_llvm_descriptor_base() local
72 if (LLVMGetTypeKind(LLVMTypeOf(binding_index)) == LLVMVectorTypeKind) in lp_llvm_descriptor_base()
73 …binding_index = LLVMBuildExtractElement(builder, binding_index, lp_build_const_int32(gallivm, 0), … in lp_llvm_descriptor_base()
75 …LLVMValueRef binding_offset = LLVMBuildMul(builder, binding_index, lp_build_const_int32(gallivm, s… in lp_llvm_descriptor_base()
|
| D | lp_bld_jit_sample.c | 464 LLVMValueRef binding_index = LLVMBuildExtractValue(builder, params->resource, 1, ""); in lp_bld_llvm_image_soa_emit_op() local
465 …LLVMValueRef inbounds = LLVMBuildICmp(builder, LLVMIntSGE, binding_index, lp_build_const_int32(gal… in lp_bld_llvm_image_soa_emit_op()
|
| /external/mesa3d/src/amd/vulkan/nir/ |
| D | radv_nir_lower_vs_inputs.c | 227 … const unsigned binding_index = s->info->vs.use_per_attribute_vb_descs ? location : attrib_binding; in lower_load_vs_input() local
228 …d desc_index = util_bitcount(s->info->vs.vb_desc_usage_mask & u_bit_consecutive(0, binding_index)); in lower_load_vs_input()
|
| D | radv_nir_apply_pipeline_layout.c | 200 unsigned binding_index = var->data.binding; in get_sampler_desc() local
204 struct radv_descriptor_set_binding_layout *binding = &layout->binding[binding_index]; in get_sampler_desc()
|
| /external/vulkan-validation-layers/layers/ |
| D | object_tracker_utils.cpp | 487 … 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()
|
| /external/mesa3d/src/virtio/vulkan/ |
| D | vn_descriptor_set.c | 439 uint32_t binding_index, in vn_pool_restore_mutable_states() argument
442 assert(layout->bindings[binding_index].type == in vn_pool_restore_mutable_states()
446 vn_get_mutable_state(pool, &layout->bindings[binding_index]); in vn_pool_restore_mutable_states()
|
| /external/mesa3d/src/gallium/frontends/lavapipe/ |
| D | lvp_descriptor_set.c | 360 for (uint32_t binding_index = 0; binding_index < layout->binding_count; binding_index++) { in lvp_descriptor_set_create() local
361 …const struct lvp_descriptor_set_binding_layout *bind_layout = &set->layout->binding[binding_index]; in lvp_descriptor_set_create()
|
| D | lvp_execute.c | 4361 for (uint32_t binding_index = 0; binding_index < set_layout->binding_count; binding_index++) { in bind_db_samplers() local
4362 … const struct lvp_descriptor_set_binding_layout *bind_layout = &set_layout->binding[binding_index]; in bind_db_samplers()
|
| /external/mesa3d/src/imagination/vulkan/ |
| D | pvr_pipeline.c | 1487 dma_desc->binding_index = attribute->binding; in pvr_graphics_pipeline_setup_vertex_dma()
|
| D | pvr_cmd_buffer.c | 3418 &state->vertex_bindings[attribute->binding_index]; in pvr_setup_vertex_buffers()
3447 &state->vertex_bindings[attribute->binding_index]; in pvr_setup_vertex_buffers()
3476 &state->vertex_bindings[attribute->binding_index]; in pvr_setup_vertex_buffers()
|