Searched refs:read_elements (Results 1 – 3 of 3) sorted by relevance
33 const size_t read_elements = (readable_elements < element_count ? in GetBufferReadRegions() local38 if (read_elements > margin) { in GetBufferReadRegions()43 *data_ptr_bytes_2 = (read_elements - margin) * buf->element_size; in GetBufferReadRegions()46 *data_ptr_bytes_1 = read_elements * buf->element_size; in GetBufferReadRegions()51 return read_elements; in GetBufferReadRegions()
145 std::vector<std::vector<Tensor>> read_elements; in TEST() local147 ReadElementsFromCheckpoint(&reader, test_prefix, &read_elements)); in TEST()148 ASSERT_EQ(elements.size(), read_elements.size()); in TEST()151 std::vector<Tensor>& read = read_elements[i]; in TEST()
631 int32_t read_elements[6]; in TEST_F() local632 uint32_t num_read_bytes = sizeof(read_elements); in TEST_F()638 MojoReadData(consumer_, &read_options, read_elements, &num_read_bytes)); in TEST_F()656 read_elements, &num_read_bytes)); in TEST_F()707 const int32_t* read_elements = static_cast<const int32_t*>(read_buffer); in TEST_F() local708 ASSERT_EQ(123, read_elements[0]); in TEST_F()728 read_elements = static_cast<const int32_t*>(read_buffer); in TEST_F()729 ASSERT_EQ(456, read_elements[0]); in TEST_F()