| /external/tensorflow/tensorflow/core/kernels/data/ |
| padded_batch_dataset_op.cc | 57 PartialTensorShape padded_shape; variable
60 padded_shape_t.NumElements(), &padded_shape));
61 padded_shapes.push_back(std::move(padded_shape));
239 const PartialTensorShape& padded_shape = variable
242 for (int dim = 0; dim < padded_shape.dims(); ++dim) {
243 if (padded_shape.dim_size(dim) == -1) {
246 batch_component_shape.AddDim(padded_shape.dim_size(dim));
255 if (element_shape.dims() != padded_shape.dims()) {
259 component_index, ": expected rank ", padded_shape.dims(),
262 for (int dim = 0; dim < padded_shape.dims(); ++dim)
[all...] |
| tensor_queue_dataset_op.cc | 183 // padded_shape[r] is known.
186 // padded_shape[r] is unknown, find the maximum across
516 PartialTensorShape padded_shape; variable
519 padded_shape_t.NumElements(), &padded_shape));
520 padded_shapes.push_back(std::move(padded_shape));
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| spacetobatch_op.cc | 52 // input according to `paddings` to produce `padded` of shape `padded_shape`.
54 std::vector<int64> padded_shape(input_shape.begin(), input_shape.end());
65 padded_shape[1 + i] += pad_start + pad_end;
82 // [padded_shape[1] / block_shape[0],
85 // padded_shape[M] / block_shape[M-1],
92 OP_REQUIRES(ctx, padded_shape[1 + i] % block_shape[i] == 0,
93 errors::InvalidArgument("padded_shape[", 1 + i,
94 "]=", padded_shape[1 + i],
98 reshaped_padded_shape[1 + i * 2] = padded_shape[1 + i] / block_shape[i];
112 // [padded_shape[1] / block_shape[0]
[all...] |
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| batch_dataset_op_test.py | 215 padded_shape = array_ops.placeholder(dtypes.int64, shape=[1])
220 4, padded_shapes=padded_shape).make_initializable_iterator())
230 padded_shape: [-1],
247 padded_shape: [25],
261 sess.run(init_op, feed_dict={padded_shape: [-1], seq_lens: [0, 0, 0, 0]})
269 sess.run(init_op, feed_dict={padded_shape: [5], seq_lens: [6, 5, 5, 5]})
275 padded_shape = array_ops.placeholder(dtypes.int64, shape=[1])
285 padded_shapes=(padded_shape, padded_shape),
296 padded_shape: [-1]
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| conv_ops_fused.cc | 711 TensorShape padded_shape variable
[all...] |
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| tensor_queue_dataset_test.py | 340 padded_shape = [-1]
345 padded_shapes=(padded_shape, padded_shape),
|
| sequence_queueing_state_saver.py | [all...] |
| /external/tensorflow/tensorflow/contrib/data/python/kernel_tests/ |
| batch_dataset_op_test.py | 529 padded_shape = [-1]
533 padded_shapes=(padded_shape, padded_shape),
|
| /external/tensorflow/tensorflow/python/ops/ |
| image_ops_impl.py | 619 padded_shape = [
623 padded.set_shape(padded_shape)
[all...] |
