| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | translate_utils.cc | 29 auto bad_consumers = b.getNamedAttr( in PopulateTfVersions() local
32 versions.bad_consumers().begin(), versions.bad_consumers().end()))); in PopulateTfVersions()
35 {producer, min_consumer, bad_consumers}))); in PopulateTfVersions()
54 auto bad_consumers = in ExtractTfVersions() local
56 if (!bad_consumers) return mlir::success(); in ExtractTfVersions()
58 for (auto bad_consumer : bad_consumers) { in ExtractTfVersions()
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | revived_types.py | 29 min_consumer_version, bad_consumers=None, setter=setattr): argument
69 if bad_consumers is None:
70 bad_consumers = []
71 self._bad_consumers = bad_consumers
83 bad_consumers=self._bad_consumers))
97 for bad_version in proto.version.bad_consumers:
|
| D | revived_types_test.py | 51 bad_consumers=[3]),
68 bad_consumers=[])))
73 self.assertEqual([3], serialized.version.bad_consumers)
85 bad_consumers=[])))
95 bad_consumers=[4, 3])))
105 bad_consumers=[])))
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | graph-version-info.pbtxt | 171 bad_consumers: 3
172 bad_consumers: 4
173 bad_consumers: 2
174 bad_consumers: 9
179 # CHECK-SAME: tf.versions = {bad_consumers = [3 : i32, 4 : i32, 2 : i32, 9 : i32], min_consumer = 7…
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | versions.proto | 22 // consumer not in bad_consumers
32 repeated int32 bad_consumers = 3; field
|
| D | versions.cc | 44 for (const int bad_consumer : versions.bad_consumers()) { in CheckVersions()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/ |
| D | function-control-ret.mlir | 3 module attributes {tf.versions = {bad_consumers = [], min_consumer = 12 : i32, producer = 29 : i32}…
|
| D | func_attr.mlir | 6 module attributes {tf.versions = {bad_consumers = [], min_consumer = 12 : i32, producer = 458 : i32…
|
| D | device-arg-retval-attr.mlir | 6 module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 121 : i32}…
|
| D | case.mlir | 3 module attributes {tf.versions = {bad_consumers = [], min_consumer = 12 : i32, producer = 486 : i32…
|
| D | parse_example.mlir | 49 module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 413 : i32}…
|
| D | switchn.mlir | 58 }) {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 126 : i32}} : () -> ()
|
| D | parse_example_v2.mlir | 3 module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 175 : i32}…
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | mlir_graph_optimization_pass_test.cc | 61 auto bad_consumers = in Run() local
66 {producer, min_consumer, bad_consumers}))); in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/ |
| D | result-sharding.mlir | 3 module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 351 : i32}…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | resource_inlining.mlir | 8 module attributes {tf.versions = {bad_consumers = [], min_consumer = 12 : i32, producer = 384 : i32…
|
| D | isolate-placer.mlir | 3 module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 130 : i32}…
|
| D | tpu-variable-runtime-reformatting.mlir | 8 module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 268 : i32}…
102 module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 268 : i32}…
168 module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 268 : i32}…
231 module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 268 : i32}…
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.cc | 237 auto* bad_consumers = versions->mutable_bad_consumers(); in Extend() local
238 const std::unordered_set<int> existing(bad_consumers->begin(), in Extend()
239 bad_consumers->end()); in Extend()
240 for (const int v : extension_def.versions().bad_consumers()) { in Extend()
242 bad_consumers->Add(v); in Extend()
|
| D | graph_constructor_test.cc | 1067 EXPECT_EQ(TF_GRAPH_DEF_VERSION - 1, graph_.versions().bad_consumers(0)); in TEST_F()
2823 EXPECT_EQ(dst.versions().bad_consumers(0), bad); in TEST_F()
2852 EXPECT_EQ(2, graph_.versions().bad_consumers(0)); in TEST_F()
2853 EXPECT_EQ(3, graph_.versions().bad_consumers(1)); in TEST_F()
2863 EXPECT_EQ(1, graph_.versions().bad_consumers(0)); in TEST_F()
2864 EXPECT_EQ(2, graph_.versions().bad_consumers(1)); in TEST_F()
2865 EXPECT_EQ(3, graph_.versions().bad_consumers(2)); in TEST_F()
2872 EXPECT_EQ(1, graph_.versions().bad_consumers(0)); in TEST_F()
2873 EXPECT_EQ(2, graph_.versions().bad_consumers(1)); in TEST_F()
2874 EXPECT_EQ(3, graph_.versions().bad_consumers(2)); in TEST_F()
|
| /external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| D | save.py | 126 producer=1, min_consumer=1, bad_consumers=[]),
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | prepare-tf-with-allowing-bf16-and-f16-type-legalization.mlir | 3 module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 268 : i32}…
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf-include-tf2xla-fallback.mlir | 16 module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 268 : i32}…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tpu_bridge_v1/ |
| D | end_to_end.mlir | 3 …/job:localhost/replica:0/task:0/device:TPU_SYSTEM:0"], tf.versions = {bad_consumers = [], min_cons…
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/ |
| D | signature_def.mlir | 108 module attributes {tf.versions = {bad_consumers = [], min_consumer = 12 : i32, producer = 554 : i32…
|