| /external/tensorflow/tensorflow/core/kernels/ |
| D | qr_op_impl.h | 159 q_shape.set_dim(ndims - 1, full_matrices_ ? m : min_size); in ComputeAsync()
164 r_shape.set_dim(ndims - 2, full_matrices_ ? m : min_size); in ComputeAsync()
179 transposed_shape.set_dim(ndims - 2, input.dim_size(ndims - 1)); in ComputeAsync()
180 transposed_shape.set_dim(ndims - 1, input.dim_size(ndims - 2)); in ComputeAsync()
|
| D | attention_ops.cc | 89 output_shape.set_dim(1, output_height); in Compute()
90 output_shape.set_dim(2, output_width); in Compute()
|
| D | split_op.cc | 157 output_shape.set_dim(split_dim, split_dim_output_size); in operator ()()
303 output_shape.set_dim(split_dim, split_dim_output_size); in Compute()
367 output_shape.set_dim(split_dim, split_dim_output_size); in Compute()
|
| D | split_v_op.cc | 219 output_shape.set_dim(split_dim, split_sizes_vec[i]); in operator ()()
383 output_shape.set_dim(split_dim, split_sizes_vec[i]); in Compute()
418 output_shape.set_dim(split_dim, split_sizes_vec[i]); in Compute()
|
| D | unique_op.cc | 121 output_shape.set_dim(axis, uniq_size); in Compute()
172 output_shape.set_dim(axis, uniq_size); in Compute()
|
| D | split_op_test.cc | 33 in_shape.set_dim(split_dim, in_shape.dim_size(split_dim) * num_split); in MakeGraph()
|
| D | reshape_op.h | 76 shape.set_dim(unknown_index, missing); in Compute()
|
| D | segment_reduction_ops.cc | 105 output_shape.set_dim(0, output_rows); in Compute()
238 output_shape.set_dim(0, 0); in ComputeAsync()
278 output_shape.set_dim(0, output_rows); in ComputeAsync()
680 output_shape.set_dim(0, output_rows); in Compute()
1038 output_shape.set_dim(0, M); in Compute()
|
| D | reshape_util.cc | 93 output_shape.set_dim(unknown_index, missing); in Reshape()
|
| D | mkl_reshape_op.cc | 113 shape.set_dim(unknown_index, missing); in Compute()
|
| D | compare_and_bitpack_op.cc | 61 output_shape.set_dim(rank - 1, input_shape.dim_size(rank - 1) / 8); in Compute()
|
| D | batch_kernels.cc | 80 output_shape.set_dim(0, output_dim0); in Concat()
155 output_shape.set_dim(0, size); in SplitCPU()
584 output_shape.set_dim(0, 0); in ProcessBatch()
1182 output_shape.set_dim(0, 0); in Compute()
|
| D | quantized_concat_op.cc | 211 output_shape.set_dim(concat_dim, output_concat_dim); in Compute()
|
| D | topk_op.cc | 77 output_shape.set_dim(input_in.dims() - 1, k); in Compute()
|
| D | deserialize_sparse_string_op.cc | 154 shape.set_dim(j, std::max(shape.dim_size(j), in Compute()
|
| D | ragged_gather_op.cc | 236 values_shape.set_dim(0, num_values); in WriteValues()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape_test.cc | 43 TEST(TensorShapeTest, set_dim) { in TEST() argument
46 s.set_dim(0, 20); in TEST()
51 s.set_dim(1, 2); in TEST()
157 s.set_dim(1, 0); in TEST()
159 s.set_dim(1, 7); in TEST()
256 void set_dim(int d, int64 size);
427 void TensorShapeOld::set_dim(int d, int64 size) { in set_dim() function in tensorflow::__anone850e9110111::TensorShapeOld
|
| D | tensor_util.cc | 67 shape.set_dim(0, total_dim0_size); in Concat()
136 shape.set_dim(0, size); in Split()
159 shape.set_dim(0, size); in Split()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reshape_op.cc | 84 shape.set_dim(unknown_index, missing); in Compile()
|
| D | split_op.cc | 186 output_shape.set_dim(split_dim, slice_size); in Compile()
|
| D | image_ops.cc | 123 channel_shape.set_dim(channel_dim, 1); in Compile()
256 channel_shape.set_dim(channel_dim, 1); in Compile()
310 channel_shape.set_dim(channel_dim, 1); in Compile()
|
| /external/tensorflow/tensorflow/contrib/resampler/kernels/ |
| D | resampler_ops.cc | 164 output_shape.set_dim(output_shape.dims() - 1, data_channels); in Compute()
366 resampler_output_shape.set_dim(resampler_output_shape.dims() - 1, in Compute()
|
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| D | pmf_to_cdf_op.cc | 57 shape.set_dim(shape.dims() - 1, shape.dim_size(shape.dims() - 1) + 1); in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 355 (*engine_input_shapes)[i].set_dim(0, smallest_batch_size); in GetCompatibleCachedEngine()
570 engine_input_shapes[i].set_dim(0, max_batch_size); in GetEngine()
|
| /external/tensorflow/tensorflow/contrib/reduce_slice_ops/kernels/ |
| D | reduce_slice_ops.cc | 147 output_shape.set_dim(axis, out_axis_dim_size); in Compute()
|