Searched defs:inner_dims (Results 1 – 2 of 2) sorted by relevance
62 def _VerifyFftMethod(self, inner_dims, complex_to_input, input_to_expected, argument
1953 absl::flat_hash_set<int64> inner_dims; in HandleSlice() local2410 std::vector<int64> inner_dims(output_min2maj.begin(), concat_dim_layout_itr); in EmitFastConcatenate() local