Searched refs:TPU_C_API_MAX_INLINED (Results 1 – 3 of 3) sorted by relevance
31 #define TPU_C_API_MAX_INLINED 6 macro185 int inlined[TPU_C_API_MAX_INLINED];193 int64_t inlined[TPU_C_API_MAX_INLINED];201 float inlined[TPU_C_API_MAX_INLINED];209 bool inlined[TPU_C_API_MAX_INLINED];231 XLA_Tile inlined[TPU_C_API_MAX_INLINED];
165 if (dst->size > TPU_C_API_MAX_INLINED) { in CreateVectorBase()193 if (dst->size > TPU_C_API_MAX_INLINED) { in CreateVector()212 const Src* src = src_list.size > TPU_C_API_MAX_INLINED ? src_list.heap in MakeSpanBase()274 if (c_shape->dimensions.size > TPU_C_API_MAX_INLINED) { in Destroy()277 if (c_shape->dynamic_dimensions.size > TPU_C_API_MAX_INLINED) { in Destroy()309 const XLA_Tile* c_tiles = c_layout->tiles.size > TPU_C_API_MAX_INLINED in FromC()321 if (c_layout->minor_to_major.size > TPU_C_API_MAX_INLINED) { in Destroy()324 if (c_layout->dim_level_types.size > TPU_C_API_MAX_INLINED) { in Destroy()327 if (c_layout->tiles.size > TPU_C_API_MAX_INLINED) { in Destroy()342 if (c_tile->dimensions.size > TPU_C_API_MAX_INLINED) { in Destroy()[all …]
667 if (args.dimensions.size > TPU_C_API_MAX_INLINED) { in set_shape()671 if (args.dynamic_dimensions.size > TPU_C_API_MAX_INLINED) { in set_shape()676 if (args.layout.minor_to_major.size > TPU_C_API_MAX_INLINED) { in set_shape()680 if (args.layout.tiles.size > TPU_C_API_MAX_INLINED) { in set_shape()