Searched refs:dtypes_ (Results 1 - 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
H A D | 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::__anon26442::TensorDatasetOp::Dataset
|
H A D | 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::__anon26444::TensorSliceDatasetOp::Dataset
|
H A D | 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_ 405 DataTypeVector dtypes_; member in class:tensorflow::__anon26443::PrependFromQueueAndPaddedBatchDataset::Iterator::TensorQueue 478 const DataTypeVector dtypes_; member in class:tensorflow::__anon26443::PrependFromQueueAndPaddedBatchDataset [all...] |
H A D | 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::__anon26437::Dataset
|
/external/tensorflow/tensorflow/core/kernels/ |
H A D | 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
|
H A D | 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()) {
|
Completed in 145 milliseconds