Home
last modified time | relevance | path

Searched refs:index_count (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h724 std::vector<int64> index_count; member
733 shape, iteration_space.index_base, iteration_space.index_count, in ForEachIndexWithStatus()
/external/mesa3d/src/panfrost/lib/
Dpan_blit.c457 cfg.index_count = vertex_count; in panfrost_load_midg()
651 cfg.index_count = vertex_count; in panfrost_load_bifrost()
Ddecode.c598 pandecode_validate_buffer(primitive.indices, primitive.index_count * size); in pandecode_primitive()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc1454 std::vector<int64> index_count; in IterationSpaceForOutputBatchIndices() local
1455 index_count.reserve(output_rank); in IterationSpaceForOutputBatchIndices()
1459 index_count.push_back(is_output_batch_dim ? output_shape.dimensions(i) : 1); in IterationSpaceForOutputBatchIndices()
1462 return {std::move(index_base), std::move(index_count), in IterationSpaceForOutputBatchIndices()
1472 std::vector<int64> index_count(output_rank, 1); in IterationSpaceForOutputOffsetIndices() local
1482 index_count[i] = slice_sizes[slice_sizes_idx++]; in IterationSpaceForOutputOffsetIndices()
1486 return {std::move(index_base), std::move(index_count), in IterationSpaceForOutputOffsetIndices()
Dhlo_evaluator_typed_visitor.h2085 std::vector<int64> index_count(updates_rank, 1);
2090 index_count[i] = updates_shape.dimensions(i);
2093 return {std::move(index_base), std::move(index_count),
2104 std::vector<int64> index_count(updates_rank, 1);
2109 index_count[i] = updates_shape.dimensions(i);
2112 return {std::move(index_base), std::move(index_count),
/external/deqp/external/openglcts/modules/gl/
Dgl3cTransformFeedbackTests.cpp389 glw::GLint index_count = 0; in testExtension1() local
391 gl.getIntegerv(GL_MAX_TRANSFORM_FEEDBACK_SEPARATE_ATTRIBS, &index_count); in testExtension1()
395 if (index_count == 0) in testExtension1()
402 gl.bindBufferRange(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 0, 16); in testExtension1()
416 m_glBindBufferOffsetEXT(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 0); in testExtension1()
429 gl.bindBufferBase(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0); in testExtension1()
443 gl.bindBufferRange(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 0, 0); in testExtension1()
457 gl.bindBufferRange(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 3, 4); in testExtension1()
470 m_glBindBufferOffsetEXT(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 3); in testExtension1()
484 gl.bindBufferRange(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 0, 3); in testExtension1()
/external/llvm-project/lldb/tools/compact-unwind/
Dcompact-unwind-dumper.c1372 uint32_t index_count = baton.unwind_header.indexCount; in print_index_sections() local
1377 while (cur_idx < index_count) { in print_index_sections()
/external/igt-gpu-tools/lib/
Dpanfrost-job.h888 u32 index_count; member
/external/mesa3d/src/gallium/frontends/lavapipe/
Dlvp_private.h721 uint32_t index_count; member
Dlvp_cmd_buffer.c725 cmd->u.draw_indexed.index_count = indexCount; in lvp_CmdDrawIndexed()
Dlvp_execute.c1850 state->info.count = cmd->u.draw_indexed.index_count; in handle_draw_indexed()
/external/mesa3d/src/gallium/drivers/panfrost/
Dpan_context.c361 cfg.index_count = info->count; in panfrost_draw_emit_tiler()
363 cfg.index_count = info->count_from_stream_output ? in panfrost_draw_emit_tiler()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc3043 uint64_t index_count = index_type.getShape()[index_vector_dim]; in fold() local
3044 for (uint64_t i = 0; i < index_count; ++i) { in fold()
/external/mesa3d/src/amd/vulkan/
Dradv_cmd_buffer.c5074 uint32_t index_count) in radv_cs_emit_draw_indexed_packet() argument
5080 radeon_emit(cmd_buffer->cs, index_count); in radv_cs_emit_draw_indexed_packet()