Searched refs:CreateSplitResourceGraph (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 | 159 Status CreateSplitResourceGraph(const TensorShape& input_shape, in CreateSplitResourceGraph() function
519 {"Resource", CreateSplitResourceGraph}}),
571 {"ResourceRanked1", 1, CreateSplitResourceGraph},
572 {"ResourceRanked2", 2, CreateSplitResourceGraph},
573 {"ResourceRanked3", 3, CreateSplitResourceGraph},
574 {"ResourceRanked4", 4, CreateSplitResourceGraph},
575 {"ResourceRanked5", 5, CreateSplitResourceGraph},
576 {"ResourceRanked6", 6, CreateSplitResourceGraph},
577 {"ResourceRanked7", 7, CreateSplitResourceGraph},
578 {"ResourceRanked8", 8, CreateSplitResourceGraph}}),
|