| /external/tensorflow/tensorflow/core/kernels/data/ |
| tensor_dataset_op.cc | 51 dtypes_.push_back(t.dtype());
62 const DataTypeVector& output_dtypes() const override { return dtypes_; }
80 b->BuildAttrValue(dtypes_, &dtypes);
128 DataTypeVector dtypes_; member in class:tensorflow::__anon39763::TensorDatasetOp::Dataset
|
| tensor_slice_dataset_op.cc | 62 dtypes_.push_back(t.dtype());
79 const DataTypeVector& output_dtypes() const override { return dtypes_; }
97 b->BuildAttrValue(dtypes_, &dtypes);
154 DataTypeVector dtypes_; member in class:tensorflow::__anon39765::TensorSliceDatasetOp::Dataset
|
| tensor_queue_dataset_op.cc | 74 dtypes_(dtypes),
130 b->BuildAttrValue(dtypes_, &output_types);
159 params.dataset->dtypes_,
170 const auto& dtypes = dataset()->dtypes_;
254 : dtypes_(dtypes),
318 if (tensors.size() != dtypes_.size()) {
322 dtypes_.size(), " tensors but tried to insert ", tensors.size());
325 if (tensors[i].dtype() != dtypes_[i]) {
331 " but queue expected type: ", DataTypeString(dtypes_[i]));
359 for (int i = 0; i < dtypes_.size(); ++i)
405 DataTypeVector dtypes_; member in class:tensorflow::__anon39764::PrependFromQueueAndPaddedBatchDataset::Iterator::TensorQueue
478 const DataTypeVector dtypes_; member in class:tensorflow::__anon39764::PrependFromQueueAndPaddedBatchDataset
[all...] |
| sparse_tensor_slice_dataset_op.cc | 37 dtypes_({DT_INT64, sparse_tensor.dtype(), DT_INT64}),
48 const DataTypeVector& output_dtypes() const override { return dtypes_; }
205 const DataTypeVector dtypes_; member in class:tensorflow::__anon39758::Dataset
|
| /external/tensorflow/tensorflow/core/kernels/ |
| save_restore_v2_ops.cc | 146 OP_REQUIRES_OK(context, context->GetAttr("dtypes", &dtypes_));
153 OP_REQUIRES(context, tensor_names.NumElements() == dtypes_.size(),
155 " tensor names, but ", dtypes_.size(),
184 shape_and_slices, dtypes_));
189 std::vector<DataType> dtypes_; member in class:tensorflow::RestoreV2
|
| map_stage_op.cc | 95 DataTypeVector dtypes_ GUARDED_BY(mu_);
166 if (index >= dtypes_.size()) {
169 "' was out of bounds '", dtypes_.size(), "'."));
214 "' was already initialized '", dtypes_.size(), "'."));
269 OptionalTuple empty(dtypes_.size());
340 : dtypes_(dtypes),
352 if (indices->NumElements() != dtypes_.size()) {
|
