Searched refs:CompileTimeConstantInput (Results 1 – 25 of 50) sorted by relevance
12
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| 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"),
99 REGISTER_XLA_OP(Name("Max").CompileTimeConstantInput("reduction_indices"),
134 REGISTER_XLA_OP(Name("Mean").CompileTimeConstantInput("reduction_indices"),
152 REGISTER_XLA_OP(Name("All").CompileTimeConstantInput("reduction_indices"),
170 REGISTER_XLA_OP(Name("Any").CompileTimeConstantInput("reduction_indices"),
|
| 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 | conv_ops.cc | 126 Name("Conv2DBackpropInput").CompileTimeConstantInput("input_sizes"),
135 Name("Conv3DBackpropInputV2").CompileTimeConstantInput("input_sizes"),
144 .CompileTimeConstantInput("input_sizes"),
185 Name("Conv2DBackpropFilter").CompileTimeConstantInput("filter_sizes"),
195 Name("Conv3DBackpropFilterV2").CompileTimeConstantInput("filter_sizes"),
204 .CompileTimeConstantInput("filter_sizes"),
|
| D | fft_ops.cc | 141 REGISTER_XLA_OP(Name("RFFT").CompileTimeConstantInput("fft_length"), RFFTOp<1>);
142 REGISTER_XLA_OP(Name("RFFT2D").CompileTimeConstantInput("fft_length"),
144 REGISTER_XLA_OP(Name("RFFT3D").CompileTimeConstantInput("fft_length"),
153 REGISTER_XLA_OP(Name("IRFFT").CompileTimeConstantInput("fft_length"),
155 REGISTER_XLA_OP(Name("IRFFT2D").CompileTimeConstantInput("fft_length"),
157 REGISTER_XLA_OP(Name("IRFFT3D").CompileTimeConstantInput("fft_length"),
|
| D | strided_slice_op.cc | 112 .CompileTimeConstantInput("begin")
113 .CompileTimeConstantInput("end")
114 .CompileTimeConstantInput("strides"),
221 .CompileTimeConstantInput("shape")
222 .CompileTimeConstantInput("begin")
223 .CompileTimeConstantInput("end")
224 .CompileTimeConstantInput("strides"),
329 .CompileTimeConstantInput("begin")
330 .CompileTimeConstantInput("end")
331 .CompileTimeConstantInput("strides"),
|
| D | reduce_window_op.cc | 133 .CompileTimeConstantInput("window_dimensions")
134 .CompileTimeConstantInput("window_strides")
135 .CompileTimeConstantInput("base_dilations")
136 .CompileTimeConstantInput("window_dilations")
137 .CompileTimeConstantInput("padding"),
|
| D | sequence_ops.cc | 115 .CompileTimeConstantInput("start")
116 .CompileTimeConstantInput("limit")
117 .CompileTimeConstantInput("delta"),
195 .CompileTimeConstantInput("start")
196 .CompileTimeConstantInput("stop")
197 .CompileTimeConstantInput("num"),
|
| D | bcast_ops.cc | 70 .CompileTimeConstantInput("s0")
71 .CompileTimeConstantInput("s1"),
125 .CompileTimeConstantInput("s0")
126 .CompileTimeConstantInput("s1"),
|
| D | segment_reduction_ops.cc | 108 Name("UnsortedSegmentSum").CompileTimeConstantInput("num_segments"),
123 Name("UnsortedSegmentProd").CompileTimeConstantInput("num_segments"),
140 Name("UnsortedSegmentMin").CompileTimeConstantInput("num_segments"),
157 Name("UnsortedSegmentMax").CompileTimeConstantInput("num_segments"),
|
| D | xla_pad_op.cc | 99 .CompileTimeConstantInput("padding_low")
100 .CompileTimeConstantInput("padding_high")
101 .CompileTimeConstantInput("padding_interior"),
|
| D | concat_op.cc | 114 REGISTER_XLA_OP(Name("Concat").CompileTimeConstantInput("concat_dim"),
118 .CompileTimeConstantInput("axis"),
202 .CompileTimeConstantInput("concat_dim")
203 .CompileTimeConstantInput("shape"),
|
| D | stateful_random_ops.cc | 289 .CompileTimeConstantInput("algorithm")
290 .CompileTimeConstantInput("shape")
322 .CompileTimeConstantInput("algorithm")
323 .CompileTimeConstantInput("shape")
355 .CompileTimeConstantInput("algorithm")
356 .CompileTimeConstantInput("shape")
|
| D | xla_select_and_scatter_op.cc | 141 .CompileTimeConstantInput("window_dimensions")
142 .CompileTimeConstantInput("window_strides")
143 .CompileTimeConstantInput("padding"),
|
| D | stateless_random_ops.cc | 104 .CompileTimeConstantInput("shape")
152 .CompileTimeConstantInput("shape")
193 .CompileTimeConstantInput("shape")
234 .CompileTimeConstantInput("shape")
|
| D | random_ops.cc | 61 REGISTER_XLA_OP(Name("RandomUniform").CompileTimeConstantInput("shape"),
237 REGISTER_XLA_OP(Name("RandomUniformInt").CompileTimeConstantInput("shape"),
266 REGISTER_XLA_OP(Name("RandomStandardNormal").CompileTimeConstantInput("shape"),
292 .CompileTimeConstantInput("shape")
|
| D | pooling_ops.cc | 191 .CompileTimeConstantInput("ksize")
192 .CompileTimeConstantInput("strides"),
364 .CompileTimeConstantInput("ksize")
365 .CompileTimeConstantInput("strides"),
461 Name("AvgPoolGrad").CompileTimeConstantInput("orig_input_shape"),
470 Name("AvgPool3DGrad").CompileTimeConstantInput("orig_input_shape"),
639 .CompileTimeConstantInput("ksize")
640 .CompileTimeConstantInput("strides"),
|
| D | split_op.cc | 97 REGISTER_XLA_OP(Name("Split").CompileTimeConstantInput("split_dim"), SplitOp);
197 .CompileTimeConstantInput("split_dim")
198 .CompileTimeConstantInput("size_splits"),
|
| D | index_ops.cc | 80 .CompileTimeConstantInput("dimension"),
91 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 | pad_op.cc | 84 REGISTER_XLA_OP(Name("Pad").CompileTimeConstantInput("paddings"), PadOp);
85 REGISTER_XLA_OP(Name("PadV2").CompileTimeConstantInput("paddings"), PadOp);
|
| D | listdiff_op.cc | 118 .CompileTimeConstantInput("x")
119 .CompileTimeConstantInput("y"),
|
| D | tensor_list_ops.cc | 126 .CompileTimeConstantInput("element_shape")
127 .CompileTimeConstantInput("num_elements"),
162 .CompileTimeConstantInput("element_shape")
163 .CompileTimeConstantInput("max_num_elements"),
308 Name("TensorListFromTensor").CompileTimeConstantInput("element_shape"),
|
| D | reverse_op.cc | 67 REGISTER_XLA_OP(Name("Reverse").CompileTimeConstantInput("dims"), ReverseOp);
119 REGISTER_XLA_OP(Name("ReverseV2").CompileTimeConstantInput("axis"),
|
| D | slice_op.cc | 116 .CompileTimeConstantInput("begin")
117 .CompileTimeConstantInput("size"),
|
12