Searched refs:CreateRoundtripTensorGraph (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 | 1217 Status CreateRoundtripTensorGraph( in CreateRoundtripTensorGraph() function
1425 {{"TensorRanked1", 1, CreateRoundtripTensorGraph},
1426 {"TensorRanked2", 2, CreateRoundtripTensorGraph},
1427 {"TensorRanked3", 3, CreateRoundtripTensorGraph},
1428 {"TensorRanked4", 4, CreateRoundtripTensorGraph},
1429 {"TensorRanked5", 5, CreateRoundtripTensorGraph},
1430 {"TensorRanked6", 6, CreateRoundtripTensorGraph},
1431 {"TensorRanked7", 7, CreateRoundtripTensorGraph},
1432 {"TensorRanked8", 8, CreateRoundtripTensorGraph},
|