Searched defs:inner_size (Results 1 – 13 of 13) sorted by relevance
45 int inner_size = 1; in Gather() local
43 int inner_size = 1; in ArgMinMax() local
1504 int64_t inner_size; in RankOneSelect() local
41 int inner_size = 0; in py_blake2s_new() local
41 int inner_size = 0; in py_blake2b_new() local
390 int inner_size = in str() local403 int inner_size = layout.order == Order::kRowMajor ? layout.cols : layout.rows; in str() local415 int inner_size = layout.order == Order::kRowMajor ? layout.cols : layout.rows; in str() local
139 int64 inner_size = 1; in Compute() local
48 const int64 inner_size = in ReverseRows() local
689 int64 inner_size = 1; in Compute() local
93 int inner_size, int last_node, int usedforsecurity) in py_blake2s_new_impl()
93 int inner_size, int last_node, int usedforsecurity) in py_blake2b_new_impl()
232 const int inner_size = shape_common.Dims(3); in RunOneLogSoftmaxTest() local
721 const int64_t inner_size = dim_size * base_inner_size; in ConstFoldConcatenateOpDense() local