Searched refs:dtypes_ (Results 1 - 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
H A Dtensor_dataset_op.cc51 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::__anon26442::TensorDatasetOp::Dataset
H A Dtensor_slice_dataset_op.cc62 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::__anon26444::TensorSliceDatasetOp::Dataset
H A Dtensor_queue_dataset_op.cc74 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_
405 DataTypeVector dtypes_; member in class:tensorflow::__anon26443::PrependFromQueueAndPaddedBatchDataset::Iterator::TensorQueue
478 const DataTypeVector dtypes_; member in class:tensorflow::__anon26443::PrependFromQueueAndPaddedBatchDataset
[all...]
H A Dsparse_tensor_slice_dataset_op.cc37 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::__anon26437::Dataset
/external/tensorflow/tensorflow/core/kernels/
H A Dsave_restore_v2_ops.cc146 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
H A Dmap_stage_op.cc95 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()) {

Completed in 145 milliseconds