| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| 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 | matrix_diag_op.cc | 245 output_shape.set_dim(diag_rank - 1, num_rows); in Compute()
248 output_shape.set_dim(diag_rank - 2, num_rows); in Compute()
249 output_shape.set_dim(diag_rank - 1, num_cols); in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape_test.cc | 44 TEST(TensorShapeTest, set_dim) { in TEST() argument
47 s.set_dim(0, 20); in TEST()
52 s.set_dim(1, 2); in TEST()
158 s.set_dim(1, 0); in TEST()
160 s.set_dim(1, 7); in TEST()
279 void set_dim(int d, int64 size);
450 void TensorShapeOld::set_dim(int d, int64 size) { in set_dim() function in tensorflow::__anon127dd62f0111::TensorShapeOld
738 shape.set_dim(0, 8); in BM_TensorShape_SetDim()
|
| D | tensor_util.cc | 72 shape.set_dim(0, total_dim0_size); in Concat()
143 shape.set_dim(0, size); in Split()
166 shape.set_dim(0, size); in Split()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | attention_ops.cc | 91 output_shape.set_dim(1, output_height); in Compute()
92 output_shape.set_dim(2, output_width); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_variant_op.cc | 60 values_shape.set_dim(0, num_values); in UnbatchRaggedZerothDim()
114 values_shape.set_dim(0, num_values); in UnbatchRaggedZerothDim()
212 zeros_shape.set_dim(0, piece_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 | 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 | reshape_op.h | 93 shape.set_dim(unknown_index, missing); in Compute()
|
| D | unique_op.cc | 165 output_shape.set_dim(axis, uniq_size); in Compute()
217 output_shape.set_dim(axis, uniq_size); in Compute()
|
| D | segment_reduction_ops_impl.h | 114 output_shape.set_dim(0, output_rows); in Compute()
249 output_shape.set_dim(0, 0); in ComputeAsync()
289 output_shape.set_dim(0, output_rows); in ComputeAsync()
497 output_shape.set_dim(0, output_rows);
869 output_shape.set_dim(0, M);
|
| D | split_op.cc | 154 output_shape.set_dim(split_dim, split_dim_output_size); in operator ()()
300 output_shape.set_dim(split_dim, split_dim_output_size); in Compute()
|
| D | reshape_util.cc | 142 output_shape.set_dim(unknown_index, missing); in ReshapeSparseTensor()
|
| D | compare_and_bitpack_op.cc | 61 output_shape.set_dim(rank - 1, input_shape.dim_size(rank - 1) / 8); in Compute()
|
| D | deserialize_sparse_string_op.cc | 156 shape.set_dim(j, std::max(shape.dim_size(j), in Compute()
|
| D | quantized_concat_op.cc | 209 output_shape.set_dim(concat_dim, output_concat_dim); in Compute()
|
| D | ragged_tensor_from_variant_op.cc | 171 component_values_shape.set_dim(0, values_size); in NestedStackRaggedTensors()
|
| D | topk_op.cc | 77 output_shape.set_dim(input_in.dims() - 1, k); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | concat_split_util.h | 73 output_shape.set_dim(0, output_dim0); in Concat()
170 output_shape.set_dim(0, size); in SplitCPU()
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 543 output_shapes[i].set_dim(split_dim, size); in Split()
591 output_shape.set_dim(dim, slice_size); in Slice()
594 output_shape.set_dim(dim, input_size - start_index); in Slice()
597 output_shape.set_dim(dim, 0); in Slice()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reshape_op.cc | 103 shape.set_dim(unknown_index, missing); in Compile()
|
| D | split_op.cc | 190 output_shape.set_dim(split_dim, slice_size); in Compile()
|
| D | strided_slice_op.cc | 77 partial_final_shape.set_dim( in EmitDynamicSlice()
93 partial_processing_shape.set_dim(i, input_shape.dim_size(i)); in EmitDynamicSlice()
435 processing_shape.set_dim(shape_spec.output_to_processing_mapping[i], in CompileAsDynamicUpdateSlice()
|
| D | image_ops.cc | 125 channel_shape.set_dim(channel_dim, 1); in Compile()
258 channel_shape.set_dim(channel_dim, 1); in Compile()
312 channel_shape.set_dim(channel_dim, 1); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_reshape_op.cc | 124 shape.set_dim(unknown_index, missing); in Compute()
|