| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | training_ops.cc | 39 TensorShape alpha_shape = ctx->InputShape(1); in Compile()
44 TensorShape delta_shape = ctx->InputShape(2); in Compile()
84 TensorShape alpha_shape = ctx->InputShape(1); in Compile()
88 TensorShape l1_shape = ctx->InputShape(2); in Compile()
92 TensorShape l2_shape = ctx->InputShape(3); in Compile()
96 TensorShape delta_shape = ctx->InputShape(4); in Compile()
136 TensorShape lr_shape = ctx->InputShape(2); in Compile()
141 TensorShape grad_shape = ctx->InputShape(3); in Compile()
147 TensorShape momentum_shape = ctx->InputShape(4); in Compile()
194 TensorShape lr_shape = ctx->InputShape(2); in Compile()
[all …]
|
| D | dynamic_slice_ops.cc | 40 const TensorShape input_shape = ctx->InputShape("input"); in Compile()
41 const TensorShape update_shape = ctx->InputShape("update"); in Compile()
42 const TensorShape index_shape = ctx->InputShape("indices"); in Compile()
75 const TensorShape input_shape = ctx->InputShape("input"); in Compile()
76 const TensorShape start_indices_shape = ctx->InputShape("start_indices"); in Compile()
77 const TensorShape size_indices_shape = ctx->InputShape("size_indices"); in Compile()
|
| D | shape_op.cc | 39 const TensorShape input_shape = ctx->InputShape(0); in Compile()
59 const TensorShape input_shape = ctx->InputShape(i); in Compile()
78 const TensorShape input_shape = ctx->InputShape(0); in Compile()
94 const TensorShape input_shape = ctx->InputShape(0); in Compile()
119 const TensorShape input_shape = ctx->InputShape("input"); in Compile()
120 const TensorShape dim_shape = ctx->InputShape("dim"); in Compile()
169 const TensorShape input_shape = ctx->InputShape(0); in Compile()
226 const TensorShape input_shape = ctx->InputShape(0); in Compile()
240 const TensorShape input_shape = ctx->InputShape(0); in Compile()
|
| D | concat_op.cc | 47 const TensorShape concat_dim_tensor_shape = ctx->InputShape(axis_index_); in Compile()
126 const TensorShape concat_dim_shape = ctx->InputShape(0); in Compile()
132 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(ctx->InputShape(i)), in Compile()
135 ctx->InputShape(i).DebugString())); in Compile()
156 const TensorShape inp0_shape = ctx->InputShape(1); in Compile()
171 const TensorShape inp_shape = ctx->InputShape(1 + i); in Compile()
|
| D | one_hot_op.cc | 33 const TensorShape indices_shape = ctx->InputShape(0); in Compile()
34 const TensorShape depth_shape = ctx->InputShape(1); in Compile()
35 const TensorShape on_value_shape = ctx->InputShape(2); in Compile()
36 const TensorShape off_value_shape = ctx->InputShape(3); in Compile()
|
| D | pooling_ops.cc | 79 const TensorShape ksize_shape = ctx->InputShape(1); in GetKernelSize()
103 const TensorShape stride_shape = ctx->InputShape(2); in GetStride()
170 const TensorShape input_shape = ctx->InputShape(0); in Compile()
223 const TensorShape input_shape = ctx->InputShape(0); in Compile()
284 const TensorShape ksize_shape = ctx->InputShape(3); in Compile()
291 const TensorShape stride_shape = ctx->InputShape(4); in Compile()
308 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile()
309 const TensorShape tensor_out_shape = ctx->InputShape(1); in Compile()
310 const TensorShape out_backprop_shape = ctx->InputShape(2); in Compile()
407 const TensorShape out_backprop_shape = ctx->InputShape(1); in Compile()
[all …]
|
| D | sequence_ops.cc | 73 const TensorShape start_in_shape = ctx->InputShape(0); in Compile()
74 const TensorShape limit_in_shape = ctx->InputShape(1); in Compile()
75 const TensorShape delta_in_shape = ctx->InputShape(2); in Compile()
125 const TensorShape start_in_shape = ctx->InputShape("start"); in Compile()
126 const TensorShape stop_in_shape = ctx->InputShape("stop"); in Compile()
127 const TensorShape num_in_shape = ctx->InputShape("num"); in Compile()
|
| D | listdiff_op.cc | 41 OP_REQUIRES(context, TensorShapeUtils::IsVector(context->InputShape(0)), in Compile()
43 context->InputShape(0).DebugString())); in Compile()
45 OP_REQUIRES(context, TensorShapeUtils::IsVector(context->InputShape(1)), in Compile()
47 context->InputShape(1).DebugString())); in Compile()
|
| D | reverse_op.cc | 37 const TensorShape x_shape = ctx->InputShape(0); in Compile()
38 const TensorShape revd_shape = ctx->InputShape(1); in Compile()
75 const TensorShape x_shape = ctx->InputShape(0); in Compile()
76 const TensorShape axes_shape = ctx->InputShape(1); in Compile()
|
| D | split_op.cc | 38 const TensorShape split_dim_shape = ctx->InputShape("split_dim"); in Compile()
39 const TensorShape input_shape = ctx->InputShape(1); in Compile()
105 const TensorShape input_shape = ctx->InputShape(0); in Compile()
106 const TensorShape index_shape = ctx->InputShape(2); in Compile()
131 const TensorShape split_size_shape = ctx->InputShape(1); in Compile()
|
| D | clip_by_value_op.cc | 29 const TensorShape shape = ctx->InputShape(0); in Compile()
30 const TensorShape min_shape = ctx->InputShape(1); in Compile()
31 const TensorShape max_shape = ctx->InputShape(2); in Compile()
|
| D | stateless_random_ops.cc | 81 TensorShape seed_shape = ctx->InputShape(1); in Compile()
120 TensorShape seed_shape = ctx->InputShape(1); in Compile()
124 TensorShape minval_shape = ctx->InputShape(2); in Compile()
128 TensorShape maxval_shape = ctx->InputShape(3); in Compile()
168 TensorShape seed_shape = ctx->InputShape(1); in Compile()
209 TensorShape seed_shape = ctx->InputShape(1); in Compile()
|
| D | transpose_op.cc | 40 const TensorShape input_shape = ctx->InputShape("x"); in Compile()
41 const TensorShape perm_tensor_shape = ctx->InputShape("perm"); in Compile()
126 FastBoundsCheck(ctx->InputShape(0).num_elements(), in Compile()
159 int size = ctx->InputShape(0).num_elements(); in Compile()
|
| D | sparse_to_dense_op.cc | 31 const TensorShape indices_shape = context->InputShape(0); in Compile()
51 const TensorShape sparse_values_shape = context->InputShape(2); in Compile()
62 const TensorShape default_value_shape = context->InputShape(3); in Compile()
|
| D | select_op.cc | 34 const TensorShape cond_shape = ctx->InputShape(0); in Compile()
35 const TensorShape then_shape = ctx->InputShape(1); in Compile()
36 const TensorShape else_shape = ctx->InputShape(2); in Compile()
|
| D | gather_op.cc | 156 auto input_shape = context->InputShape(0); in Compile()
158 auto indices_shape = context->InputShape(1); in Compile()
161 const TensorShape axis_shape = context->InputShape(2); in Compile()
206 TensorShape params_shape = context->InputShape(0); in Compile()
207 TensorShape indices_shape = context->InputShape(1); in Compile()
|
| D | pad_op.cc | 33 const TensorShape input_shape = ctx->InputShape("input"); in Compile()
34 const TensorShape pad_shape = ctx->InputShape("paddings"); in Compile()
74 ctx, TensorShapeUtils::IsScalar(ctx->InputShape("constant_values")), in Compile()
|
| D | xla_conv_op.cc | 44 const TensorShape lhs_shape = context->InputShape(0); in Compile()
45 const TensorShape rhs_shape = context->InputShape(1); in Compile()
46 const TensorShape padding_shape = context->InputShape("padding"); in Compile()
|
| D | lrn_ops.cc | 38 const TensorShape in_shape = ctx->InputShape(0); in Compile()
90 const TensorShape in_grads_shape = ctx->InputShape(0); in Compile()
91 const TensorShape in_image_shape = ctx->InputShape(1); in Compile()
92 const TensorShape out_image_shape = ctx->InputShape(2); in Compile()
|
| D | softmax_op.cc | 40 const TensorShape logits_shape = ctx->InputShape(0); in Compile()
148 const TensorShape logits_shape = ctx->InputShape(0); in Compile()
149 const TensorShape labels_shape = ctx->InputShape(1); in Compile()
181 const TensorShape logits_shape = ctx->InputShape(0); in Compile()
182 const TensorShape labels_shape = ctx->InputShape(1); in Compile()
|
| D | slice_op.cc | 40 const TensorShape input_shape = ctx->InputShape(0); in Compile()
41 const TensorShape begin_tensor_shape = ctx->InputShape(1); in Compile()
42 const TensorShape size_tensor_shape = ctx->InputShape(2); in Compile()
|
| D | matrix_band_part_op.cc | 32 const TensorShape input_shape = context->InputShape(0); in Compile()
39 const TensorShape num_lower_in_shape = context->InputShape(1); in Compile()
44 const TensorShape num_upper_in_shape = context->InputShape(2); in Compile()
|
| D | bias_ops.cc | 42 const TensorShape input_shape = ctx->InputShape(0); in Compile()
43 const TensorShape bias_shape = ctx->InputShape(1); in Compile()
91 const TensorShape out_backprop_shape = ctx->InputShape(0); in Compile()
|
| D | diag_op.cc | 86 const TensorShape input_shape = ctx->InputShape(0); in Compile()
126 const TensorShape input_shape = ctx->InputShape(0); in Compile()
170 const TensorShape input_shape = ctx->InputShape(0); in Compile()
197 const TensorShape input_shape = ctx->InputShape(0); in Compile()
|
| D | einsum_op.cc | 41 const TensorShape a_shape = ctx->InputShape(0); in Compile()
42 const TensorShape b_shape = ctx->InputShape(1); in Compile()
|