Searched refs:assertTensorSpec (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/federated-compute/fcp/artifact_building/ |
| H A D | graph_helpers_test.py | 38 def assertTensorSpec(self, tensor, name, shape, dtype): member in EmbedDataLogicTest
55 self.assertTensorSpec(token_placeholder, 'data_token:0', [], tf.string)
57 self.assertTensorSpec(data_values[0], 'data_0/Identity:0', [], tf.variant)
79 self.assertTensorSpec(token_placeholder, 'data_token:0', [], tf.string)
82 self.assertTensorSpec(data_values[0], 'data_0/Identity:0', [], tf.variant)
83 self.assertTensorSpec(data_values[1], 'data_1/Identity:0', [], tf.variant)
103 self.assertTensorSpec(token_placeholder, 'data_token:0', [], tf.string)
105 self.assertTensorSpec(data_values[0], 'data_0/Identity:0', [], tf.variant)
114 self.assertTensorSpec(token_placeholder, 'data_token:0', [], tf.string)
116 self.assertTensorSpec(data_values[0], 'data_0/Identity:0', [], tf.variant)
[all …]
|