| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | dense_to_sparse_batch_dataset_op.cc | 52 PartialTensorShape row_shape; in MakeDataset() local
55 row_shape_t->NumElements(), &row_shape)); in MakeDataset()
61 *output = new Dataset<T>(ctx, batch_size, row_shape, input); \ in MakeDataset()
82 const PartialTensorShape& row_shape, const DatasetBase* input) in Dataset() argument
85 row_shape_(row_shape), in Dataset()
134 std::vector<int64> row_shape; in AsGraphDefInternal() local
135 row_shape.reserve( in AsGraphDefInternal()
138 row_shape.emplace_back(row_shape_.dim_size(i)); in AsGraphDefInternal()
139 TF_RETURN_IF_ERROR(b->AddVector(row_shape, &row_shape_node)); in AsGraphDefInternal()
165 const PartialTensorShape& row_shape = in GetNextInternal() local
[all …]
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | batching.py | 29 def dense_to_sparse_batch(batch_size, row_shape): argument
71 return batching.dense_to_sparse_batch(batch_size, row_shape)
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | batching.py | 156 def dense_to_sparse_batch(batch_size, row_shape): argument
201 return _DenseToSparseBatchDataset(dataset, batch_size, row_shape)
451 def __init__(self, input_dataset, batch_size, row_shape): argument
460 self._row_shape = row_shape
468 row_shape=convert.partial_shape_to_tensor(self._row_shape),
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ExperimentalDenseToSparseBatchDataset.pbtxt | 18 name: "row_shape"
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | factorization_ops.py | 882 row_shape = [
888 row_shape = [
898 new_sp_shape = (array_ops.concat([row_shape, col_shape], 0)
900 array_ops.concat([col_shape, row_shape], 0))
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 2198 TensorShape dense_shape, row_shape; in FastParseSequenceExample() local
2202 if (!c.shape.AsTensorShape(&row_shape) || in FastParseSequenceExample()
2204 (expected_max_elements / row_shape.num_elements()) * in FastParseSequenceExample()
2205 row_shape.num_elements()) { in FastParseSequenceExample()
2209 int64 expected_max_rows = expected_max_elements / row_shape.num_elements(); in FastParseSequenceExample()
2285 if (num_added != row_shape.num_elements()) { in FastParseSequenceExample()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.pbtxt | 109 argspec: "args=[\'batch_size\', \'row_shape\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.raw_ops.pbtxt | 1061 …argspec: "args=[\'input_dataset\', \'batch_size\', \'row_shape\', \'output_types\', \'output_shape…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.pbtxt | 109 argspec: "args=[\'batch_size\', \'row_shape\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.raw_ops.pbtxt | 1061 …argspec: "args=[\'input_dataset\', \'batch_size\', \'row_shape\', \'output_types\', \'output_shape…
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 2095 const RuntimeShape row_shape({1, batches, output_height, output_width}); in DilatedIm2col()
2100 {1, 1, row_shape.FlatSize(), col_shape.FlatSize()}); in DilatedIm2col()
2108 int row_offset = Offset(row_shape, 0, batch, out_y, out_x); in DilatedIm2col()
6275 const RuntimeShape row_shape({1, batches, output_height, output_width}); in TransposeIm2col()
6280 {1, 1, row_shape.FlatSize(), col_shape.FlatSize()}); in TransposeIm2col()
6308 int row_offset = Offset(row_shape, 0, batch, out_y, out_x); in TransposeIm2col()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1972 const RuntimeShape row_shape({1, batches, output_height, output_width}); in DilatedIm2col()
1977 {1, 1, row_shape.FlatSize(), col_shape.FlatSize()}); in DilatedIm2col()
1985 int row_offset = Offset(row_shape, 0, batch, out_y, out_x); in DilatedIm2col()
6444 const RuntimeShape row_shape({1, batches, output_height, output_width}); in TransposeIm2col()
6449 {1, 1, row_shape.FlatSize(), col_shape.FlatSize()}); in TransposeIm2col()
6477 int row_offset = Offset(row_shape, 0, batch, out_y, out_x); in TransposeIm2col()
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 29771 …tchDataset(scope *Scope, input_dataset tf.Output, batch_size tf.Output, row_shape tf.Output, outpu…
29779 input_dataset, batch_size, row_shape,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 23240 name: "row_shape"
23272 name: "row_shape"
|
| D | ops_history.v1.pbtxt | 23611 name: "row_shape"
23643 name: "row_shape"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 11157 name: "row_shape"
|