Searched defs:row_shape (Results 1 - 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
H A D | dense_to_sparse_batch_dataset_op.cc | 49 OP_REQUIRES_OK(ctx, ctx->input("row_shape", &row_shape_t)); 51 errors::InvalidArgument("row_shape must be a vector")); 52 PartialTensorShape row_shape; variable 55 row_shape_t->NumElements(), &row_shape)); 61 *output = new Dataset<T>(ctx, batch_size, row_shape, input); \ 82 const PartialTensorShape& row_shape, const DatasetBase* input) 85 row_shape_(row_shape), 125 std::vector<int64> row_shape; variable 126 row_shape.reserve( 129 row_shape 81 Dataset(OpKernelContext* ctx, int64 batch_size, const PartialTensorShape& row_shape, const DatasetBase* input) argument 152 const PartialTensorShape& row_shape = variable [all...] |
Completed in 265 milliseconds