| /external/tensorflow/tensorflow/core/framework/ |
| tensor_util.cc | 61 shape.set_dim(0, total_dim0_size);
130 shape.set_dim(0, size);
153 shape.set_dim(0, size);
|
| tensor_shape_test.cc | 43 TEST(TensorShapeTest, set_dim) {
46 s.set_dim(0, 20);
51 s.set_dim(1, 2);
157 s.set_dim(1, 0);
159 s.set_dim(1, 7);
249 void set_dim(int d, int64 size);
420 void TensorShapeOld::set_dim(int d, int64 size) { function in class:tensorflow::__anon39597::TensorShapeOld
|
| /external/lisa/libs/utils/android/ |
| screen.py | 95 def set_dim(target, auto=True): member in class:Screen
131 Screen.set_dim(target)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| attention_ops.cc | 65 output_shape.set_dim(1, output_height);
66 output_shape.set_dim(2, output_width);
|
| qr_op_impl.h | 156 q_shape.set_dim(ndims - 1, full_matrices_ ? m : min_size);
161 r_shape.set_dim(ndims - 2, full_matrices_ ? m : min_size);
176 transposed_shape.set_dim(ndims - 2, input.dim_size(ndims - 1));
177 transposed_shape.set_dim(ndims - 1, input.dim_size(ndims - 2));
|
| reshape_op.h | 76 shape.set_dim(unknown_index, missing);
|
| split_op_test.cc | 33 in_shape.set_dim(split_dim, in_shape.dim_size(split_dim) * num_split);
|
| split_op.cc | 162 output_shape.set_dim(split_dim, split_dim_output_size);
262 output_shape.set_dim(split_dim, split_dim_output_size);
326 output_shape.set_dim(split_dim, split_dim_output_size);
|
| split_v_op.cc | 241 output_shape.set_dim(split_dim, split_sizes_vec[i]);
339 output_shape.set_dim(split_dim, split_sizes_vec[i]);
374 output_shape.set_dim(split_dim, split_sizes_vec[i]);
|
| reshape_util.cc | 93 output_shape.set_dim(unknown_index, missing);
|
| mkl_reshape_op.cc | 86 shape.set_dim(unknown_index, missing);
211 shape.set_dim(unknown_index, missing);
|
| unique_op.cc | 121 output_shape.set_dim(axis, uniq_size);
171 output_shape.set_dim(axis, uniq_size);
|
| batch_kernels.cc | 78 output_shape.set_dim(0, output_dim0);
154 output_shape.set_dim(0, size);
342 output_shape.set_dim(0, 0);
[all...] |
| compare_and_bitpack_op.cc | 61 output_shape.set_dim(rank - 1, input_shape.dim_size(rank - 1) / 8);
|
| segment_reduction_ops.cc | 103 output_shape.set_dim(0, output_rows);
236 output_shape.set_dim(0, 0);
276 output_shape.set_dim(0, output_rows);
676 output_shape.set_dim(0, output_rows);
[all...] |
| concat_op.cc | 128 output_shape.set_dim(axis, output_concat_dim);
|
| quantized_concat_op.cc | 211 output_shape.set_dim(concat_dim, output_concat_dim);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| reshape_op.cc | 82 shape.set_dim(unknown_index, missing);
|
| conv_ops.cc | 44 expanded_shape.set_dim(num_dims - 1, shape.dim_size(num_dims - 2) *
140 implicit_broadcast_filter_shape.set_dim(
142 implicit_broadcast_filter_shape.set_dim(
|
| image_ops.cc | 111 channel_shape.set_dim(channel_dim, 1);
236 channel_shape.set_dim(channel_dim, 1);
286 channel_shape.set_dim(channel_dim, 1);
|
| gather_op.cc | 59 slice_shape.set_dim(axis + i, 1);
|
| reverse_sequence_op.cc | 136 slice_shape.set_dim(batch_dim_, 1);
|
| split_op.cc | 201 output_shape.set_dim(split_dim, slice_size);
|
| /external/tensorflow/tensorflow/contrib/resampler/kernels/ |
| resampler_ops.cc | 164 output_shape.set_dim(output_shape.dims() - 1, data_channels);
366 resampler_output_shape.set_dim(resampler_output_shape.dims() - 1,
|
| /external/tensorflow/tensorflow/contrib/reduce_slice_ops/kernels/ |
| reduce_slice_ops.cc | 142 output_shape.set_dim(axis, out_axis_dim_size);
|
