Searched refs:CreateConcatTensorGraph (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 | 790 Status CreateConcatTensorGraph(absl::Span<const TensorShape> input_shapes, in CreateConcatTensorGraph() function
1142 {{"Tensor", CreateConcatTensorGraph},
1189 {{"TensorRanked1", 1, CreateConcatTensorGraph},
1190 {"TensorRanked2", 2, CreateConcatTensorGraph},
1191 {"TensorRanked3", 3, CreateConcatTensorGraph},
1192 {"TensorRanked4", 4, CreateConcatTensorGraph},
1193 {"TensorRanked5", 5, CreateConcatTensorGraph},
1194 {"TensorRanked6", 6, CreateConcatTensorGraph},
1195 {"TensorRanked7", 7, CreateConcatTensorGraph},
1196 {"TensorRanked8", 8, CreateConcatTensorGraph},
|