Searched defs:ParameterizedMetadataTest (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | sharding_propagation_test.cc | 71 class ParameterizedMetadataTest class
201 TEST_P(ParameterizedMetadataTest, ShardingMetadataFromInstruction) { in TEST_P() argument
302 TEST_P(ParameterizedMetadataTest, ElementwiseOperationForwardPass) { in TEST_P() argument
333 TEST_P(ParameterizedMetadataTest, ElementwiseOperationBackwardPass) { in TEST_P() argument
400 TEST_P(ParameterizedMetadataTest, BroadcastBackwardPass) { in TEST_P() argument
430 TEST_P(ParameterizedMetadataTest, Broadcast1DBackwardNoChange) { in TEST_P() argument
493 TEST_P(ParameterizedMetadataTest, BroadcastMerge) { in TEST_P() argument
524 TEST_P(ParameterizedMetadataTest, BroadcastUser) { in TEST_P() argument
592 TEST_P(ParameterizedMetadataTest, MaximalReduceForwardPass) { in TEST_P() argument
628 TEST_P(ParameterizedMetadataTest, ShardedReduceForwardPass) { in TEST_P() argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | tpu_rewrite_device_util_test.cc | 91 struct ParameterizedMetadataTest struct
96 TEST_P(ParameterizedMetadataTest, BadMetadata) { in TEST_P() argument
|