Searched refs:CreateSplitTensorGraph (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 | 132 Status CreateSplitTensorGraph(const TensorShape& input_shape, in CreateSplitTensorGraph() function
518 {{"Tensor", CreateSplitTensorGraph},
563 {{"TensorRanked1", 1, CreateSplitTensorGraph},
564 {"TensorRanked2", 2, CreateSplitTensorGraph},
565 {"TensorRanked3", 3, CreateSplitTensorGraph},
566 {"TensorRanked4", 4, CreateSplitTensorGraph},
567 {"TensorRanked5", 5, CreateSplitTensorGraph},
568 {"TensorRanked6", 6, CreateSplitTensorGraph},
569 {"TensorRanked7", 7, CreateSplitTensorGraph},
570 {"TensorRanked8", 8, CreateSplitTensorGraph},
|