HomeSort by relevance Sort by last modified time
    Searched defs:padded_shape (Results 1 - 3 of 3) sorted by null

  /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/core/kernels/
conv_ops_fused.cc 711 TensorShape padded_shape variable
    [all...]

Completed in 95 milliseconds