| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | dense_to_sparse_batch_dataset_op.cc | 50 PartialTensorShape row_shape; in MakeDataset() local
53 row_shape_t->NumElements(), &row_shape)); in MakeDataset()
59 *output = new Dataset<T>(ctx, batch_size, row_shape, input); \ in MakeDataset()
80 const PartialTensorShape& row_shape, const DatasetBase* input) in Dataset() argument
83 row_shape_(row_shape), in Dataset()
142 std::vector<int64> row_shape; in AsGraphDefInternal() local
143 row_shape.reserve( in AsGraphDefInternal()
146 row_shape.emplace_back(row_shape_.dim_size(i)); in AsGraphDefInternal()
147 TF_RETURN_IF_ERROR(b->AddVector(row_shape, &row_shape_node)); in AsGraphDefInternal()
173 const PartialTensorShape& row_shape = in GetNextInternal() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | logdet.cc | 53 TF_ASSIGN_OR_RETURN(auto row_shape, in LogDet()
55 rows = Iota(a.builder(), row_shape, row_shape.rank() - 1); in LogDet()
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | batching.py | 102 def dense_to_sparse_batch(batch_size, row_shape): argument
146 return _DenseToSparseBatchDataset(dataset, batch_size, row_shape)
300 def __init__(self, input_dataset, batch_size, row_shape): argument
309 self._row_shape = row_shape
317 row_shape=convert.partial_shape_to_tensor(self._row_shape),
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ExperimentalDenseToSparseBatchDataset.pbtxt | 12 name: "row_shape"
44 name: "row_shape"
|
| D | DenseToSparseBatchDataset.pbtxt | 12 name: "row_shape"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ExperimentalDenseToSparseBatchDataset.pbtxt | 12 name: "row_shape"
44 name: "row_shape"
|
| D | DenseToSparseBatchDataset.pbtxt | 12 name: "row_shape"
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | im2col_utils.h | 150 const RuntimeShape row_shape({1, batches, output_height, output_width}); in DilatedIm2col()
155 {1, 1, row_shape.FlatSize(), col_shape.FlatSize()}); in DilatedIm2col()
165 int row_offset = Offset(row_shape, 0, batch, out_y, out_x); in DilatedIm2col()
|
| D | optimized_ops.h | 5609 const RuntimeShape row_shape({1, batches, output_height, output_width}); in TransposeIm2col()
5614 {1, 1, row_shape.FlatSize(), col_shape.FlatSize()}); in TransposeIm2col()
5642 int row_offset = Offset(row_shape, 0, batch, out_y, out_x); in TransposeIm2col()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_DenseToSparseBatchDataset.pbtxt | 18 name: "row_shape"
|
| D | api_def_ExperimentalDenseToSparseBatchDataset.pbtxt | 18 name: "row_shape"
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 2493 TensorShape dense_shape, row_shape; in ParseSequenceDenseFeatures() local
2496 if (!c.shape.AsTensorShape(&row_shape) || in ParseSequenceDenseFeatures()
2498 (expected_max_elements / row_shape.num_elements()) * in ParseSequenceDenseFeatures()
2499 row_shape.num_elements()) { in ParseSequenceDenseFeatures()
2500 PartialTensorShape total_shape = row_shape; in ParseSequenceDenseFeatures()
2508 int64 expected_max_rows = expected_max_elements / row_shape.num_elements(); in ParseSequenceDenseFeatures()
2595 if (num_added != row_shape.num_elements()) { in ParseSequenceDenseFeatures()
2600 " but output shape: ", row_shape.DebugString()); in ParseSequenceDenseFeatures()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.pbtxt | 113 argspec: "args=[\'batch_size\', \'row_shape\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.raw_ops.pbtxt | 1173 …argspec: "args=[\'input_dataset\', \'batch_size\', \'row_shape\', \'output_types\', \'output_shape…
1429 …argspec: "args=[\'input_dataset\', \'batch_size\', \'row_shape\', \'output_types\', \'output_shape…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.pbtxt | 141 argspec: "args=[\'batch_size\', \'row_shape\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.raw_ops.pbtxt | 1173 …argspec: "args=[\'input_dataset\', \'batch_size\', \'row_shape\', \'output_types\', \'output_shape…
1429 …argspec: "args=[\'input_dataset\', \'batch_size\', \'row_shape\', \'output_types\', \'output_shape…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 11333 …tchDataset(scope *Scope, input_dataset tf.Output, batch_size tf.Output, row_shape tf.Output, outpu…
11341 input_dataset, batch_size, row_shape,
46629 …tchDataset(scope *Scope, input_dataset tf.Output, batch_size tf.Output, row_shape tf.Output, outpu…
46637 input_dataset, batch_size, row_shape,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 12501 name: "row_shape"
14808 name: "row_shape"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 23240 name: "row_shape"
23272 name: "row_shape"
|