Searched refs:input_buffer_sizes_in_bytes_ (Results 1 – 2 of 2) sorted by relevance
1194 input_buffer_sizes_in_bytes_.reserve(computation_layout.parameter_count()); in TfrtCpuExecutable()1196 input_buffer_sizes_in_bytes_.push_back( in TfrtCpuExecutable()1200 input_buffer_sizes_in_bytes_.reserve( in TfrtCpuExecutable()1204 input_buffer_sizes_in_bytes_.push_back(ShapeUtil::ByteSizeOf( in TfrtCpuExecutable()1285 if (input_buffers.size() != input_buffer_sizes_in_bytes_.size()) { in CheckBufferCompatibilities()1289 input_buffers.size(), input_buffer_sizes_in_bytes_.size()); in CheckBufferCompatibilities()1293 if (input_buffer_sizes_in_bytes_[i] != buffer->Buffers()[0]->size()) { in CheckBufferCompatibilities()1297 i, input_buffer_sizes_in_bytes_[i], buffer->Buffers()[0]->size()); in CheckBufferCompatibilities()
579 std::vector<int64_t> input_buffer_sizes_in_bytes_; variable