Searched refs:expected_buffer_size (Results 1 – 3 of 3) sorted by relevance
257 size_t expected_buffer_size = tfrt::GetHostSize(element_type) * n_elem; in VerifyBufferDesc() local258 if (LLVM_UNLIKELY(expected_buffer_size != buffer.size())) { in VerifyBufferDesc()261 buffer.size(), " vs expected ", expected_buffer_size, "."); in VerifyBufferDesc()
399 int expected_buffer_size = participants_[0].source_buffers[0].size(); in RunCollectiveOp() local409 CHECK_EQ(p.destination_buffers[i].size(), expected_buffer_size); in RunCollectiveOp()410 CHECK_EQ(p.source_buffers[i].size(), expected_buffer_size); in RunCollectiveOp()436 sender.source_buffers[i].opaque(), expected_buffer_size); in RunCollectiveOp()
2136 const int expected_buffer_size = tensor_type_ == TensorType_INT8 in TEST_P() local2139 EXPECT_EQ(buffer->data.size(), expected_buffer_size); in TEST_P()