Searched refs:CompileTimeConstantInput (Results 1 – 25 of 59) sorted by relevance
123
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_conv_op.cc | 92 .CompileTimeConstantInput("window_strides")
93 .CompileTimeConstantInput("lhs_dilation")
94 .CompileTimeConstantInput("rhs_dilation")
95 .CompileTimeConstantInput("feature_group_count")
96 .CompileTimeConstantInput("padding"),
|
| D | reduction_ops.cc | 44 REGISTER_XLA_OP(Name("Sum").CompileTimeConstantInput("reduction_indices"),
63 REGISTER_XLA_OP(Name("Prod").CompileTimeConstantInput("reduction_indices"),
81 REGISTER_XLA_OP(Name("Min").CompileTimeConstantInput("reduction_indices"),
113 REGISTER_XLA_OP(Name("Max").CompileTimeConstantInput("reduction_indices"),
148 REGISTER_XLA_OP(Name("Mean").CompileTimeConstantInput("reduction_indices"),
166 REGISTER_XLA_OP(Name("All").CompileTimeConstantInput("reduction_indices"),
184 REGISTER_XLA_OP(Name("Any").CompileTimeConstantInput("reduction_indices"),
|
| D | matrix_diag_ops.cc | 332 .CompileTimeConstantInput("k")
333 .CompileTimeConstantInput("num_rows")
334 .CompileTimeConstantInput("num_cols")
335 .CompileTimeConstantInput("padding_value"),
338 .CompileTimeConstantInput("k")
339 .CompileTimeConstantInput("num_rows")
340 .CompileTimeConstantInput("num_cols")
341 .CompileTimeConstantInput("padding_value"),
446 .CompileTimeConstantInput("k")
447 .CompileTimeConstantInput("padding_value"),
[all …]
|
| D | fft_ops.cc | 144 .CompileTimeConstantInput("fft_length"),
149 .CompileTimeConstantInput("fft_length"),
154 .CompileTimeConstantInput("fft_length"),
166 .CompileTimeConstantInput("fft_length"),
171 .CompileTimeConstantInput("fft_length"),
176 .CompileTimeConstantInput("fft_length"),
|
| D | conv_ops.cc | 136 .CompileTimeConstantInput("input_sizes")
146 .CompileTimeConstantInput("input_sizes")
156 .CompileTimeConstantInput("input_sizes")
198 .CompileTimeConstantInput("filter_sizes")
209 .CompileTimeConstantInput("filter_sizes")
219 .CompileTimeConstantInput("filter_sizes")
|
| D | reduce_window_op.cc | 132 .CompileTimeConstantInput("window_dimensions")
133 .CompileTimeConstantInput("window_strides")
134 .CompileTimeConstantInput("base_dilations")
135 .CompileTimeConstantInput("window_dilations")
136 .CompileTimeConstantInput("padding"),
|
| D | stateful_random_ops.cc | 253 .CompileTimeConstantInput("algorithm")
254 .CompileTimeConstantInput("shape")
292 .CompileTimeConstantInput("algorithm")
293 .CompileTimeConstantInput("shape")
338 .CompileTimeConstantInput("algorithm")
339 .CompileTimeConstantInput("shape")
372 .CompileTimeConstantInput("algorithm")
373 .CompileTimeConstantInput("shape")
405 .CompileTimeConstantInput("algorithm")
406 .CompileTimeConstantInput("shape")
[all …]
|
| D | bcast_ops.cc | 70 .CompileTimeConstantInput("s0")
71 .CompileTimeConstantInput("s1"),
125 .CompileTimeConstantInput("s0")
126 .CompileTimeConstantInput("s1"),
|
| D | xla_pad_op.cc | 91 .CompileTimeConstantInput("padding_low")
92 .CompileTimeConstantInput("padding_high")
93 .CompileTimeConstantInput("padding_interior"),
|
| D | concat_op.cc | 114 REGISTER_XLA_OP(Name("Concat").CompileTimeConstantInput("concat_dim"),
118 .CompileTimeConstantInput("axis"),
204 .CompileTimeConstantInput("concat_dim")
205 .CompileTimeConstantInput("shape"),
|
| D | strided_slice_op.cc | 363 .CompileTimeConstantInput("begin")
364 .CompileTimeConstantInput("end")
365 .CompileTimeConstantInput("strides"),
598 .CompileTimeConstantInput("shape")
599 .CompileTimeConstantInput("begin")
600 .CompileTimeConstantInput("end")
601 .CompileTimeConstantInput("strides"),
715 .CompileTimeConstantInput("begin")
716 .CompileTimeConstantInput("end")
717 .CompileTimeConstantInput("strides"),
[all …]
|
| D | stateless_random_ops_v2.cc | 244 .CompileTimeConstantInput("shape")
245 .CompileTimeConstantInput("alg")
305 .CompileTimeConstantInput("shape")
306 .CompileTimeConstantInput("alg")
351 .CompileTimeConstantInput("shape")
352 .CompileTimeConstantInput("alg")
405 .CompileTimeConstantInput("shape")
406 .CompileTimeConstantInput("alg")
459 .CompileTimeConstantInput("shape")
460 .CompileTimeConstantInput("alg")
|
| D | segment_reduction_ops.cc | 140 Name("UnsortedSegmentSum").CompileTimeConstantInput("num_segments"),
155 Name("UnsortedSegmentProd").CompileTimeConstantInput("num_segments"),
172 Name("UnsortedSegmentMin").CompileTimeConstantInput("num_segments"),
189 Name("UnsortedSegmentMax").CompileTimeConstantInput("num_segments"),
|
| D | sequence_ops.cc | 133 .CompileTimeConstantInput("start")
134 .CompileTimeConstantInput("limit")
135 .CompileTimeConstantInput("delta"),
177 REGISTER_XLA_OP(Name("LinSpace").CompileTimeConstantInput("num"), LinSpaceOp);
|
| D | xla_select_and_scatter_op.cc | 140 .CompileTimeConstantInput("window_dimensions")
141 .CompileTimeConstantInput("window_strides")
142 .CompileTimeConstantInput("padding"),
|
| D | random_ops.cc | 67 REGISTER_XLA_OP(Name("RandomUniform").CompileTimeConstantInput("shape"),
252 REGISTER_XLA_OP(Name("RandomUniformInt").CompileTimeConstantInput("shape"),
281 REGISTER_XLA_OP(Name("RandomStandardNormal").CompileTimeConstantInput("shape"),
312 .CompileTimeConstantInput("shape")
359 .CompileTimeConstantInput("shape")
|
| D | pad_op.cc | 84 REGISTER_XLA_OP(Name("Pad").CompileTimeConstantInput("paddings"), PadOp);
85 REGISTER_XLA_OP(Name("PadV2").CompileTimeConstantInput("paddings"), PadOp);
|
| D | index_ops.cc | 89 REGISTER_XLA_OP(Name("ArgMax").CompileTimeConstantInput("dimension"),
100 REGISTER_XLA_OP(Name("ArgMin").CompileTimeConstantInput("dimension"),
|
| D | batchtospace_op.cc | 162 .CompileTimeConstantInput("block_shape")
163 .CompileTimeConstantInput("crops"),
186 REGISTER_XLA_OP(Name("BatchToSpace").CompileTimeConstantInput("crops"),
|
| D | spacetobatch_op.cc | 164 .CompileTimeConstantInput("paddings")
165 .CompileTimeConstantInput("block_shape"),
188 REGISTER_XLA_OP(Name("SpaceToBatch").CompileTimeConstantInput("paddings"),
|
| D | split_op.cc | 97 REGISTER_XLA_OP(Name("Split").CompileTimeConstantInput("split_dim"), SplitOp);
201 .CompileTimeConstantInput("split_dim")
202 .CompileTimeConstantInput("size_splits"),
|
| D | pooling_ops.cc | 199 .CompileTimeConstantInput("ksize")
200 .CompileTimeConstantInput("strides"),
385 .CompileTimeConstantInput("ksize")
386 .CompileTimeConstantInput("strides"),
481 Name("AvgPoolGrad").CompileTimeConstantInput("orig_input_shape"),
490 Name("AvgPool3DGrad").CompileTimeConstantInput("orig_input_shape"),
662 .CompileTimeConstantInput("ksize")
663 .CompileTimeConstantInput("strides"),
|
| D | listdiff_op.cc | 118 .CompileTimeConstantInput("x")
119 .CompileTimeConstantInput("y"),
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | cross_replica_ops.cc | 127 REGISTER_XLA_OP(Name("AllToAll").CompileTimeConstantInput("group_assignment"),
132 .CompileTimeConstantInput("source_target_pairs"),
135 Name("CrossReplicaSum").CompileTimeConstantInput("group_assignment"),
|
| D | image_resize_ops.cc | 138 .CompileTimeConstantInput("size")
143 .CompileTimeConstantInput("size")
148 .CompileTimeConstantInput("size")
|
123