| /external/tensorflow/tensorflow/compiler/xla/ |
| sparse_index_array.cc | 32 CHECK_LE(index_count(), max_indices_);
45 int64 SparseIndexArray::index_count() const { function in class:xla::SparseIndexArray
85 int64 num_indices = index_count();
|
| sparse_index_array.h | 72 int64 index_count() const;
128 int64 num_elements = index_count();
|
| shape_util.h | 640 std::vector<int64> index_count; member in struct:xla::ShapeUtil::IndexIterationSpace
649 shape, iteration_space.index_base, iteration_space.index_count,
[all...] |
| literal.h | 371 ? sparse_indices()->index_count()
[all...] |
| literal.cc | [all...] |
| literal_util.h | 418 CHECK_EQ(num_elements, indices.index_count());
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_evaluator.cc | 785 std::vector<int64> index_count; local
786 index_count.reserve(output_rank);
790 index_count.push_back(is_output_batch_dim ? output_shape.dimensions(i) : 1);
793 return {std::move(index_base), std::move(index_count),
803 std::vector<int64> index_count(output_rank, 1);
813 index_count[i] = slice_sizes[slice_sizes_idx++];
817 return {std::move(index_base), std::move(index_count),
[all...] |
| hlo_evaluator_typed_visitor.h | [all...] |
| /external/deqp/external/openglcts/modules/gl/ |
| gl3cTransformFeedbackTests.cpp | 382 glw::GLint index_count = 0; local
384 gl.getIntegerv(GL_MAX_TRANSFORM_FEEDBACK_SEPARATE_ATTRIBS, &index_count);
388 if (index_count == 0)
395 gl.bindBufferRange(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 0, 16);
409 m_glBindBufferOffsetEXT(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 0);
422 gl.bindBufferBase(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0);
436 gl.bindBufferRange(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 0, 0);
450 gl.bindBufferRange(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 3, 4);
463 m_glBindBufferOffsetEXT(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 3);
477 gl.bindBufferRange(GL_TRANSFORM_FEEDBACK_BUFFER, index_count, m_buffer_0, 0, 3)
[all...] |
| /external/mesa3d/src/amd/vulkan/ |
| radv_cmd_buffer.c | [all...] |
