Searched defs:row_shape (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
H A Ddense_to_sparse_batch_dataset_op.cc49 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