Searched refs:TPU_C_API_MAX_INLINED (Results 1 – 2 of 2) sorted by relevance
31 #define TPU_C_API_MAX_INLINED 6 macro185 int64_t inlined[TPU_C_API_MAX_INLINED];193 bool inlined[TPU_C_API_MAX_INLINED];205 XLA_Tile inlined[TPU_C_API_MAX_INLINED];
159 if (dst->size > TPU_C_API_MAX_INLINED) { in CopyVectorBase()179 if (dst->size > TPU_C_API_MAX_INLINED) { in CopyVector()198 const Src* src = src_list.size > TPU_C_API_MAX_INLINED ? src_list.heap in MakeSpanBase()253 if (c_shape->dimensions.size > TPU_C_API_MAX_INLINED) { in Free()256 if (c_shape->dynamic_dimensions.size > TPU_C_API_MAX_INLINED) { in Free()282 const XLA_Tile* c_tiles = c_layout->tiles.size > TPU_C_API_MAX_INLINED in FromC()293 if (c_layout->minor_to_major.size > TPU_C_API_MAX_INLINED) { in Free()296 if (c_layout->tiles.size > TPU_C_API_MAX_INLINED) { in Free()312 if (c_tile->dimensions.size > TPU_C_API_MAX_INLINED) { in Free()