Searched refs:num_elements_per_minibatch (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example.cc | 73 const size_t num_elements_per_minibatch, in FillAndCopyVarLen() argument
103 data += num_elements_per_minibatch; in FillAndCopyVarLen()
569 const size_t num_elements_per_minibatch = num_elements / batch_size; in FastParseExampleLite() local
573 num_elements_per_minibatch, config, in FastParseExampleLite()
578 FillAndCopyVarLen<float>(d, num_elements, num_elements_per_minibatch, in FastParseExampleLite()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 999 const size_t num_elements_per_minibatch, const Config& config, in FillAndCopyVarLen() argument
1032 data += num_elements_per_minibatch; in FillAndCopyVarLen()
1390 const size_t num_elements_per_minibatch = num_elements / batch_size; in FastParseExample() local
1394 FillAndCopyVarLen<int64>(d, num_elements, num_elements_per_minibatch, in FastParseExample()
1399 FillAndCopyVarLen<float>(d, num_elements, num_elements_per_minibatch, in FastParseExample()
1404 FillAndCopyVarLen<tstring>(d, num_elements, num_elements_per_minibatch, in FastParseExample()
|