MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GetNumElements

Function GetNumElements

tensorflow/compiler/xla/tools/driver.cc:325–331  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

323}
324
325int GetNumElements(const ArrayShape& shape) {
326 int num_elements = 1;
327 for (int dim : shape.dimensions) {
328 num_elements *= dim;
329 }
330 return num_elements;
331}
332
333template <typename T, typename = std::enable_if_t<std::is_integral<T>::value>>
334void FillIntT(void* buffer, int num_elements) {

Callers 4

matchAndRewriteMethod · 0.85
FillFunction · 0.85
DisplayFunction · 0.85
mainFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected