Searched refs:suffix_size (Results 1 – 3 of 3) 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);
774 const int suffix_size = suffix.dims(); in EndsWith() local775 if (shape.dims() < suffix_size) return false; in EndsWith()776 for (int i = 0; i < suffix_size; ++i) { in EndsWith()777 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()