Searched refs:suffix_size (Results 1 – 6 of 6) sorted by relevance
87 Eigen::Index suffix_size = output->dimensions()[2];95 if (suffix_size == 1) {108 const Eigen::Index d0 = i / suffix_size;109 const Eigen::Index d1 = i - (d0 * suffix_size);116 d.parallelFor(prefix_size * suffix_size, cost * suffix_size, func);
96 size_t suffix_size = request_size - aligned_size; in Allocate() local99 suffix_size); in Allocate()100 request_size -= suffix_size; in Allocate()
208 size_t suffix_size = request_size - size; in Allocate() local209 CHECK(Free(aligned_base + size, suffix_size)); in Allocate()210 request_size -= suffix_size; in Allocate()
382 size_t suffix_size = request_size - size; in Allocate() local383 CHECK(Free(aligned_base + size, suffix_size)); in Allocate()384 request_size -= suffix_size; in Allocate()
580 const int suffix_size = suffix.dims(); in EndsWith() local581 if (shape.dims() < suffix_size) return false; in EndsWith()582 for (int i = 0; i < suffix_size; ++i) { in EndsWith()583 if (shape.dim_size(shape.dims() - suffix_size + i) != suffix.dim_size(i)) { in EndsWith()
417 size_t suffix_size; in rolling_append() local424 suffix_size = strlen(suffix); in rolling_append()426 size = current_size + suffix_size; in rolling_append()427 if (size < current_size || size < suffix_size) in rolling_append()