Searched refs:elements_per_stride (Results 1 – 7 of 7) sorted by relevance
46 std::size_t elements_per_stride) in Dense()53 elements_per_stride(elements_per_stride) {} in Dense()64 std::size_t elements_per_stride; member
158 std::vector<std::size_t>* elements_per_stride);170 GetDenseShapes(dense_shapes, &variable_length, &elements_per_stride));198 std::vector<std::size_t> elements_per_stride; member225 GetDenseShapes(dense_shapes, &variable_length, &elements_per_stride)); in Init()235 std::vector<std::size_t> elements_per_stride; member
254 size_t elements_per_stride, in AddDenseFeature() argument263 new_feature.elements_per_stride = elements_per_stride; in AddDenseFeature()
635 std::vector<std::size_t>* elements_per_stride) { in GetDenseShapes() argument665 elements_per_stride->push_back(dense_shape.num_elements()); in GetDenseShapes()
662 const std::size_t num_elements = config.dense[d].elements_per_stride; in FastParseSerializedExample()715 const std::size_t num_elements = config.dense[d].elements_per_stride; in FastParseSerializedExample()1373 const size_t stride_size = config.dense[d].elements_per_stride; in FastParseExample()1376 DCHECK_EQ(max_num_features % config.dense[d].elements_per_stride, 0); in FastParseExample()1587 const std::size_t num_elements = config.dense[d].elements_per_stride; in FastParseSingleExample()1632 is_dense ? config.dense[d].elements_per_stride : 1; in FastParseSingleExample()
182 const std::size_t num_elements = config.dense[d].elements_per_stride; in FastParseSerializedExample()230 const std::size_t num_elements = config.dense[d].elements_per_stride; in FastParseSerializedExample()433 int elements_per_stride) { in CopyToBuffer() argument437 for (int j = 0; j < elements_per_stride; ++j) { in CopyToBuffer()445 for (int j = 0; j < elements_per_stride; ++j) { in CopyToBuffer()551 const size_t stride_size = config.dense[d].elements_per_stride; in FastParseExampleLite()554 DCHECK_EQ(max_num_features % config.dense[d].elements_per_stride, 0); in FastParseExampleLite()598 const int elements_per_stride = config.dense[d].elements_per_stride; in FastParseExampleLite() local605 for (int j = 0; j < elements_per_stride; ++j) { in FastParseExampleLite()636 elements_per_stride); in FastParseExampleLite()[all …]
219 attrs_.elements_per_stride[d]); in MakeConfig()358 attrs_.elements_per_stride[d]}); in Compute()