Searched refs:number_of_indices (Results 1 – 3 of 3) sorted by relevance
115 int number_of_indices = max_index + 1; in Compile() local117 if (number_of_indices == 0) { in Compile()132 std::vector<int32> src_input_vector(number_of_indices); in Compile()133 std::vector<int32> src_slice_vector(number_of_indices); in Compile()134 std::vector<bool> src_index_used(number_of_indices); in Compile()147 OP_REQUIRES(ctx, index_used_count == number_of_indices, in Compile()178 std::vector<xla::XlaOp> to_concat(number_of_indices); in Compile()179 for (int index_num = 0; index_num < number_of_indices; index_num++) { in Compile()
2269 auto number_of_indices = start_index_shapes.size(); in InferDynamicSliceShape() local2272 (number_of_indices >= 1 && start_index_shapes[0].rank() == 1)) { in InferDynamicSliceShape()2273 if (number_of_indices != 1) { in InferDynamicSliceShape()2276 number_of_indices); in InferDynamicSliceShape()2313 if (operand_shape.rank() != number_of_indices) { in InferDynamicSliceShape()2317 number_of_indices, operand_shape.rank(), in InferDynamicSliceShape()2321 if (number_of_indices > 0) { in InferDynamicSliceShape()2375 auto number_of_indices = start_index_shapes.size(); in InferDynamicUpdateSliceShape() local2378 (number_of_indices >= 1 && start_index_shapes[0].rank() == 1)) { in InferDynamicUpdateSliceShape()2379 if (number_of_indices != 1) { in InferDynamicUpdateSliceShape()[all …]
2590 const glw::GLuint number_of_indices = 128 * m_gl_max_geometry_atomic_counter_buffers_ext_value; in iterate() local2712 gl.drawArrays(GL_POINTS, 0 /*starting index*/, number_of_indices); in iterate()2717 for (glw::GLuint vertex_id = 0; vertex_id < number_of_indices; ++vertex_id) in iterate()