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

Function IsBatchMatchesForAllValues

tensorflow/lite/delegates/gpu/gl/api.cc:374–382  ·  view source on GitHub ↗

@return true if all tensors have same batch value.

Source from the content-addressed store, hash-verified

372
373// @return true if all tensors have same batch value.
374bool IsBatchMatchesForAllValues(const GraphFloat32& model) {
375 const int32_t b = model.values()[0]->tensor.shape.b;
376 for (auto value : model.values()) {
377 if (value->tensor.shape.b != b) {
378 return false;
379 }
380 }
381 return true;
382}
383
384bool IsOpenGl31OrAbove(const GpuInfo& gpu_info) {
385 return (gpu_info.major_version == 3 && gpu_info.minor_version >= 1) ||

Callers 1

CompileFunction · 0.85

Calls 1

valuesMethod · 0.45

Tested by

no test coverage detected