Searched refs:paddings_vec (Results 1 – 2 of 2) sorted by relevance
2513 std::vector<int64> paddings_vec; in TEST_F() local2519 paddings_vec.push_back(low_pad_size); in TEST_F()2520 paddings_vec.push_back(pad_size - low_pad_size); in TEST_F()2525 paddings.CopyFrom(AsIntTensor(tpaddings, paddings_vec), in TEST_F()
2122 const std::vector<int64> paddings_vec = GetFlatInt64(*paddings_t); in SpaceToBatchShapeHelper() local2124 const int64 pad_start = paddings_vec[dim * 2], in SpaceToBatchShapeHelper()2125 pad_end = paddings_vec[dim * 2 + 1]; in SpaceToBatchShapeHelper()