Searched refs:elements_per_stride (Results 1 – 5 of 5) sorted by relevance
192 elements_per_stride.push_back(dense_shape.num_elements()); in Init()203 std::vector<std::size_t> elements_per_stride; variable258 elements_per_stride.push_back(dense_shape.num_elements()); in Init()269 std::vector<std::size_t> elements_per_stride; variable
254 size_t elements_per_stride, in AddDenseFeature() argument263 new_feature.elements_per_stride = elements_per_stride; in AddDenseFeature()
52 std::size_t elements_per_stride; member
615 const std::size_t num_elements = config.dense[d].elements_per_stride; in FastParseSerializedExample()668 const std::size_t num_elements = config.dense[d].elements_per_stride; in FastParseSerializedExample()1174 const size_t stride_size = config.dense[d].elements_per_stride; in FastParseExample()1177 DCHECK_EQ(max_num_features % config.dense[d].elements_per_stride, 0); in FastParseExample()1364 const std::size_t num_elements = config.dense[d].elements_per_stride; in FastParseSingleExample()1406 is_dense ? config.dense[d].elements_per_stride : 1; in FastParseSingleExample()
129 attrs_.elements_per_stride[d]}); in Compute()231 attrs_.elements_per_stride[d]}); in Compute()