Home
last modified time | relevance | path

Searched refs:SetIsStateful (Results 1 – 25 of 57) sorted by relevance

123

/external/tensorflow/tensorflow/core/ops/
Dtpu_configuration_ops.cc82 .SetIsStateful()
108 .SetIsStateful()
134 .SetIsStateful()
149 .SetIsStateful()
163 .SetIsStateful()
180 .SetIsStateful()
195 .SetIsStateful()
199 .SetIsStateful()
Dnccl_ops.cc31 .SetIsStateful()
42 .SetIsStateful()
51 .SetIsStateful()
74 .SetIsStateful()
99 .SetIsStateful()
107 .SetIsStateful()
129 .SetIsStateful()
Drandom_ops.cc28 .SetIsStateful()
40 .SetIsStateful()
55 .SetIsStateful()
69 .SetIsStateful()
87 .SetIsStateful()
97 .SetIsStateful()
105 .SetIsStateful()
125 .SetIsStateful()
149 .SetIsStateful()
167 .SetIsStateful()
Dio_ops.cc66 .SetIsStateful()
92 .SetIsStateful()
136 .SetIsStateful()
149 .SetIsStateful()
172 .SetIsStateful()
198 .SetIsStateful()
214 .SetIsStateful()
265 .SetIsStateful()
272 .SetIsStateful()
280 .SetIsStateful()
[all …]
Ddata_flow_ops.cc190 .SetIsStateful()
203 .SetIsStateful()
213 .SetIsStateful()
223 .SetIsStateful()
233 .SetIsStateful()
243 .SetIsStateful()
253 .SetIsStateful()
263 .SetIsStateful()
269 .SetIsStateful()
423 .SetIsStateful()
[all …]
Dcandidate_sampling_ops.cc58 .SetIsStateful();
72 .SetIsStateful();
86 .SetIsStateful();
100 .SetIsStateful();
120 .SetIsStateful();
133 .SetIsStateful();
Dcollective_ops.cc32 .SetIsStateful()
43 .SetIsStateful()
54 .SetIsStateful()
64 .SetIsStateful()
Dsendrecv_ops.cc29 .SetIsStateful()
53 .SetIsStateful()
77 .SetIsStateful()
104 .SetIsStateful()
Dtpu_outfeed_ops.cc28 .SetIsStateful()
34 .SetIsStateful()
42 .SetIsStateful()
50 .SetIsStateful()
Dfunction_ops.cc26 .SetIsStateful()
42 .SetIsStateful()
58 .SetIsStateful()
73 .SetIsStateful()
Dtpu_infeed_ops.cc29 .SetIsStateful()
39 .SetIsStateful();
48 .SetIsStateful();
54 .SetIsStateful()
Dfunctional_ops.cc62 .SetIsStateful()
75 .SetIsStateful()
114 .SetIsStateful()
143 .SetIsStateful()
171 .SetIsStateful()
203 .SetIsStateful()
272 .SetIsStateful()
Dlogging_ops.cc28 .SetIsStateful()
39 .SetIsStateful()
51 .SetIsStateful()
138 .SetIsStateful()
Dscoped_allocator_ops.cc29 .SetIsStateful()
61 .SetIsStateful()
94 .SetIsStateful()
Ddataset_ops.cc40 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
52 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
65 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
82 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
341 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
407 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
427 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
450 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
470 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
517 .SetIsStateful()
[all …]
Dexperimental_dataset_ops.cc68 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
110 .SetIsStateful()
155 .SetIsStateful()
244 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
298 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
369 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
498 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
514 .SetIsStateful()
Dlookup_ops.cc343 .SetIsStateful()
353 .SetIsStateful()
363 .SetIsStateful()
373 .SetIsStateful()
387 .SetIsStateful()
398 .SetIsStateful()
418 .SetIsStateful()
433 .SetIsStateful()
Dcudnn_rnn_ops.cc69 .SetIsStateful()
111 .SetIsStateful()
156 .SetIsStateful()
209 .SetIsStateful()
246 .SetIsStateful()
284 .SetIsStateful()
Dtpu_host_compute_ops.cc28 .SetIsStateful()
49 .SetIsStateful()
Dscript_ops.cc27 .SetIsStateful()
44 .SetIsStateful()
/external/tensorflow/tensorflow/contrib/bigtable/ops/
Dbigtable_ops.cc62 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
71 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
78 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
88 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
103 .SetIsStateful() // TODO(b/123753214): Source dataset ops must be marked
/external/tensorflow/tensorflow/contrib/memory_stats/ops/
Dmemory_stats_ops.cc22 .SetIsStateful()
26 .SetIsStateful()
30 .SetIsStateful()
/external/tensorflow/tensorflow/compiler/jit/ops/
Dxla_ops.cc36 .SetIsStateful()
66 .SetIsStateful()
91 .SetIsStateful()
/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass_test.cc147 REGISTER_OP("Input").Output("o: float").SetIsStateful();
148 REGISTER_OP("InputList").Output("o: N * float").Attr("N: int").SetIsStateful();
149 REGISTER_OP("HalfInput").Output("o: half").SetIsStateful();
150 REGISTER_OP("Int32Input").Output("o: int32").SetIsStateful();
151 REGISTER_OP("DoubleInput").Output("o: double").SetIsStateful();
152 REGISTER_OP("_MklInput").Output("o: uint8").SetIsStateful();
156 .SetIsStateful();
157 REGISTER_OP("Output2").Input("i: float").Input("i1: float").SetIsStateful();
158 REGISTER_OP("Output").Input("i: float").SetIsStateful();
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc247 .SetIsStateful()
291 .SetIsStateful()
409 .SetIsStateful()
469 .SetIsStateful()
494 .SetIsStateful()

123