Searched refs:CreateConcatResourceGraph (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | sharding_util_ops_test.cc | 820 Status CreateConcatResourceGraph( in CreateConcatResourceGraph() function
1143 {"InitializedResource", CreateConcatResourceGraph<true>},
1144 {"UninitializedResource", CreateConcatResourceGraph<false>}}),
1197 {"InitializedResourceRanked1", 1, CreateConcatResourceGraph<true>},
1198 {"InitializedResourceRanked2", 2, CreateConcatResourceGraph<true>},
1199 {"InitializedResourceRanked3", 3, CreateConcatResourceGraph<true>},
1200 {"InitializedResourceRanked4", 4, CreateConcatResourceGraph<true>},
1201 {"InitializedResourceRanked5", 5, CreateConcatResourceGraph<true>},
1202 {"InitializedResourceRanked6", 6, CreateConcatResourceGraph<true>},
1203 {"InitializedResourceRanked7", 7, CreateConcatResourceGraph<true>},
[all …]
|