Searched defs:shape_idx (Results 1 – 5 of 5) sorted by relevance
71 int shape_idx, gain_idx; in quantize_py() local
732 size_t shape_idx = rank_max - 1 - idx; in Broadcast() local753 size_t shape_idx = rank_max - 1 - idx; in Broadcast() local758 size_t shape_idx = rank_max - 1 - idx; in Broadcast() local
465 int (*c)[16], float (*cn)[16], int *shape_idx, int *gain_idx) in quantize()
394 for (auto& shape_idx : outputs_in_alternate_mem) { in GetInstructionElapsedDueToMemory() local
4093 int shape_idx = -1) { in TEST_P()