| /external/tensorflow/tensorflow/compiler/xla/ |
| D | sparse_index_array.cc | 32 CHECK_LE(index_count(), max_indices_); in SparseIndexArray()
45 int64 SparseIndexArray::index_count() const { in index_count() function in xla::SparseIndexArray
85 int64 num_indices = index_count(); in Validate()
|
| D | sparse_index_array.h | 72 int64 index_count() const;
128 int64 num_elements = index_count(); in SortWithValues()
|
| D | shape_util.h | 640 std::vector<int64> index_count; member
649 shape, iteration_space.index_base, iteration_space.index_count, in ForEachIndexWithStatus()
|
| D | literal.h | 371 ? sparse_indices()->index_count() in element_count()
970 int64 last_element = p.sparse_indices()->index_count(); in AppendSparseElement()
1066 CHECK_EQ(num_elements, indices.index_count()); in PopulateSparse()
|
| D | literal.cc | 969 CHECK_LT(sparse_element_number, p.sparse_indices()->index_count()); in GetSparseIndex()
1033 int64 num_elements = sparse_indices()->index_count(); in SortSparseElementsInternal()
1193 return sparse_indices()->index_count(); in sparse_element_count()
1981 const int64 index_count = proto.sparse_indices_size() / subshape().rank(); in CopyFromProto() local
1982 sparse_indices()->Resize(index_count); in CopyFromProto()
|
| D | literal_util.h | 418 CHECK_EQ(num_elements, indices.index_count()); in CreateSparse()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 785 std::vector<int64> index_count; in IterationSpaceForOutputBatchIndices() local
786 index_count.reserve(output_rank); in IterationSpaceForOutputBatchIndices()
790 index_count.push_back(is_output_batch_dim ? output_shape.dimensions(i) : 1); in IterationSpaceForOutputBatchIndices()
793 return {std::move(index_base), std::move(index_count), in IterationSpaceForOutputBatchIndices()
803 std::vector<int64> index_count(output_rank, 1); in IterationSpaceForOutputOffsetIndices() local
813 index_count[i] = slice_sizes[slice_sizes_idx++]; in IterationSpaceForOutputOffsetIndices()
817 return {std::move(index_base), std::move(index_count), in IterationSpaceForOutputOffsetIndices()
|
| D | hlo_evaluator_typed_visitor.h | 2062 std::vector<int64> index_count(updates_rank, 1); in IterationSpaceForUpdateScatterIndices()
2067 index_count[i] = updates_shape.dimensions(i); in IterationSpaceForUpdateScatterIndices()
2070 return {std::move(index_base), std::move(index_count), in IterationSpaceForUpdateScatterIndices()
2081 std::vector<int64> index_count(updates_rank, 1); in IterationSpaceForUpdateWindowIndices()
2086 index_count[i] = updates_shape.dimensions(i); in IterationSpaceForUpdateWindowIndices()
2089 return {std::move(index_base), std::move(index_count), in IterationSpaceForUpdateWindowIndices()
|
| /external/deqp/external/openglcts/modules/gl/ |
| D | gl3cTransformFeedbackTests.cpp | 382 glw::GLint index_count = 0; in testExtension1() local
384 gl.getIntegerv(GL_MAX_TRANSFORM_FEEDBACK_SEPARATE_ATTRIBS, &index_count); in testExtension1()
388 if (index_count == 0) in testExtension1()
395 gl.bindBufferRange(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 0, 16); in testExtension1()
409 m_glBindBufferOffsetEXT(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 0); in testExtension1()
422 gl.bindBufferBase(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0); in testExtension1()
436 gl.bindBufferRange(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 0, 0); in testExtension1()
450 gl.bindBufferRange(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 3, 4); in testExtension1()
463 m_glBindBufferOffsetEXT(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 3); in testExtension1()
477 gl.bindBufferRange(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 0, 3); in testExtension1()
|
| /external/mesa3d/src/amd/vulkan/ |
| D | radv_cmd_buffer.c | 3193 uint32_t index_count) in radv_cs_emit_draw_indexed_packet() argument
3199 radeon_emit(cmd_buffer->cs, index_count); in radv_cs_emit_draw_indexed_packet()
|