Searched refs:StackArray (Results 1 – 8 of 8) sorted by relevance
531 struct StackArray { struct555 inline std::tuple<dim3, dim3, StackArray<int64_t>> check_shape_and_partition_( in check_shape_and_partition_()582 StackArray<int64_t> jagged_dims_tensor; in check_shape_and_partition_()597 const StackArray<int64_t>& jagged_dims, in walk_down_tensor_storage_tree_()598 const StackArray<index_t*>& x_offsets) { in walk_down_tensor_storage_tree_()643 StackArray<index_t*> x_offsets, in __launch_bounds__()646 StackArray<int64_t> jagged_dims, in __launch_bounds__()720 StackArray<int64_t> jagged_dims_tensor; in jagged_dense_elementwise_dense_output_()732 StackArray<index_t*> x_offset_ptrs; \ in jagged_dense_elementwise_dense_output_()765 StackArray<index_t*> x_offset_ptrs; \[all …]
52 class StackArray { class146 StackArray::kMaxCapacity >= in RunConversion()150 StackArray::RunWithCapacity( in RunConversion()255 static_assert(StackArray::kMaxCapacity >= in RunConversion()258 StackArray::RunWithCapacity( in RunConversion()
49 class StackArray { class137 static_cast<int>(StackArray::kMaxCapacity) >= in RunConversion()141 StackArray::RunWithCapacity( in RunConversion()245 static_assert(StackArray::kMaxCapacity >= in RunConversion()248 StackArray::RunWithCapacity((Limits::digits + exp + 31) / 32, in RunConversion()