Searched defs:ParameterizedMetadataTest (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | sharding_propagation_test.cc | 58 class ParameterizedMetadataTest class
139 TEST_P(ParameterizedMetadataTest, ShardingMetadataFromInstruction) { in TEST_P() argument
240 TEST_P(ParameterizedMetadataTest, ElementwiseOperationForwardPass) { in TEST_P() argument
271 TEST_P(ParameterizedMetadataTest, ElementwiseOperationBackwardPass) { in TEST_P() argument
303 TEST_P(ParameterizedMetadataTest, BroadcastForwardPass) { in TEST_P() argument
333 TEST_P(ParameterizedMetadataTest, BroadcastBackwardPass) { in TEST_P() argument
363 TEST_P(ParameterizedMetadataTest, BroadcastForwardPartial) { in TEST_P() argument
395 TEST_P(ParameterizedMetadataTest, BroadcastMerge) { in TEST_P() argument
426 TEST_P(ParameterizedMetadataTest, BroadcastUser) { in TEST_P() argument
456 TEST_P(ParameterizedMetadataTest, BroadcastUserPartial) { in TEST_P() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util_test.cc | 90 struct ParameterizedMetadataTest struct
95 TEST_P(ParameterizedMetadataTest, BadMetadata) { in TEST_P() argument
|