| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | tensor_dataset_op.cc | 41 dtypes_.reserve(tensors_.size()); in Dataset()
44 dtypes_.push_back(t.dtype()); in Dataset()
55 const DataTypeVector& output_dtypes() const override { return dtypes_; } in output_dtypes()
91 b->BuildAttrValue(dtypes_, &dtypes); in AsGraphDefInternal()
145 DataTypeVector dtypes_; member in tensorflow::data::TensorDatasetOp::Dataset
|
| D | tensor_slice_dataset_op.cc | 43 dtypes_.push_back(t.dtype()); in Dataset()
68 const DataTypeVector& output_dtypes() const override { return dtypes_; } in output_dtypes()
104 b->BuildAttrValue(dtypes_, &dtypes); in AsGraphDefInternal()
170 DataTypeVector dtypes_; member in tensorflow::data::TensorSliceDatasetOp::Dataset
|
| D | sparse_tensor_slice_dataset_op.cc | 37 dtypes_({DT_INT64, sparse_tensor.dtype(), DT_INT64}), in Dataset()
48 const DataTypeVector& output_dtypes() const override { return dtypes_; } in output_dtypes()
220 const DataTypeVector dtypes_; member in tensorflow::data::__anon1b965c8e0111::Dataset
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | outfeed_ops.cc | 57 OP_REQUIRES_OK(ctx, ctx->GetAttr("dtypes", &dtypes_)); in OutfeedEnqueueTupleOp()
69 TensorShapeToXLAShape(dtypes_[i], shapes[i], &xla_shape)); in Compile()
83 DataTypeVector dtypes_; member in tensorflow::__anon6f0d6ee40111::OutfeedEnqueueTupleOp
|
| D | infeed_op.cc | 117 OP_REQUIRES_OK(ctx, ctx->GetAttr("dtypes", &dtypes_)); in InfeedDequeueTupleOp()
121 TensorShapeToXLAShape(dtypes_[i], shapes_[i], &xla_shape)); in InfeedDequeueTupleOp()
152 DataTypeVector dtypes_; member in tensorflow::__anondc5d46500111::InfeedDequeueTupleOp
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | snapshot_util.cc | 165 dtypes_(dtypes) {} in CustomWriter()
187 simple_tensor_mask_.reserve(dtypes_.size()); in Initialize()
188 for (const auto& dtype : dtypes_) { in Initialize()
365 dtypes_(dtypes), in Dataset()
369 const DataTypeVector& output_dtypes() const override { return dtypes_; } in output_dtypes()
409 dataset()->version_, dataset()->dtypes_, &reader_)); in Initialize()
463 dataset()->version_, dataset()->dtypes_, &reader_); in AdvanceToNextFile()
477 const DataTypeVector dtypes_; member in tensorflow::data::snapshot_util::Reader::Dataset
487 dtypes_.push_back(DT_VARIANT); in NestedDataset()
493 const DataTypeVector& output_dtypes() const override { return dtypes_; } in output_dtypes()
[all …]
|
| D | snapshot_util.h | 156 const DataTypeVector dtypes_; variable
226 const DataTypeVector dtypes_; variable
277 const DataTypeVector dtypes_; variable
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | infeed_ops.cc | 298 OP_REQUIRES_OK(ctx, ctx->GetAttr("dtypes", &dtypes_)); in PrelinearizeTupleOp()
300 ctx, shapes_.size() == dtypes_.size(), in PrelinearizeTupleOp()
303 shapes_.size(), ", dtypes length = ", dtypes_.size())); in PrelinearizeTupleOp()
309 TensorShapeToXLAShape(dtypes_[i], shapes_[i], &xla_shape)); in PrelinearizeTupleOp()
330 OP_REQUIRES(ctx, input_tensor.dtype() == dtypes_[i], in Compute()
333 "; expected ", DataType_Name(dtypes_[i]), ", got ", in Compute()
369 DataTypeVector dtypes_; member in tensorflow::__anon86ada7cf0111::PrelinearizeTupleOp
453 OP_REQUIRES_OK(ctx, ctx->GetAttr("dtypes", &dtypes_)); in TpuInfeedEnqueueTupleOp()
455 ctx, shapes_.size() == dtypes_.size(), in TpuInfeedEnqueueTupleOp()
462 TensorShapeToXLAShape(dtypes_[i], shapes_[i], &xla_shape)); in TpuInfeedEnqueueTupleOp()
|
| D | outfeed_ops.cc | 70 OP_REQUIRES_OK(ctx, ctx->GetAttr("dtypes", &dtypes_)); in TpuOutfeedDequeueTupleOp()
72 ctx, shapes_.size() == dtypes_.size(), in TpuOutfeedDequeueTupleOp()
79 TensorShapeToXLAShape(dtypes_[i], shapes_[i], &xla_shape)); in TpuOutfeedDequeueTupleOp()
|
| D | infeed_ops.h | 59 DataTypeVector dtypes_; variable
|
| D | outfeed_ops.h | 60 DataTypeVector dtypes_; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_v2_ops.cc | 167 OP_REQUIRES_OK(context, context->GetAttr("dtypes", &dtypes_)); in RestoreV2()
174 OP_REQUIRES(context, tensor_names.NumElements() == dtypes_.size(), in Compute()
176 " tensor names, but ", dtypes_.size(), in Compute()
205 shape_and_slices, dtypes_)); in Compute()
210 std::vector<DataType> dtypes_; member in tensorflow::RestoreV2
|
| D | map_stage_op.cc | 95 DataTypeVector dtypes_ TF_GUARDED_BY(mu_);
166 if (index >= dtypes_.size()) { in check_index()
169 "' was out of bounds '", dtypes_.size(), "'.")); in check_index()
215 "' was already initialized '", dtypes_.size(), "'.")); in check_index_uninitialized()
271 OptionalTuple empty(dtypes_.size()); in put_incomplete()
342 : dtypes_(dtypes), in StagingMap()
354 if (indices->NumElements() != dtypes_.size()) { in put()
|
| D | resource_variable_ops.cc | 165 OP_REQUIRES_OK(c, c->GetAttr("dtypes", &dtypes_)); in ReadVariablesOp()
166 OP_REQUIRES(c, n == dtypes_.size(), in ReadVariablesOp()
169 " vs. ", dtypes_.size(), ")")); in ReadVariablesOp()
173 std::vector<core::RefCountPtr<Var>> variables(dtypes_.size()); in Compute()
174 std::vector<const ResourceHandle*> handles(dtypes_.size()); in Compute()
175 for (size_t i = 0; i < dtypes_.size(); ++i) { in Compute()
194 for (size_t i = 0; i < dtypes_.size(); ++i) { in Compute()
199 OP_REQUIRES(ctx, dtypes_[i] == variables[i]->tensor()->dtype(), in Compute()
203 " with wrong dtype. Expected ", DataTypeString(dtypes_[i]), in Compute()
|
| D | resource_variable_ops.h | 58 DataTypeVector dtypes_;
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 266 OP_REQUIRES_OK(ctx, ctx->GetAttr("dtypes", &dtypes_)); in XRTAllocateFromTensorOp()
273 ctx, tf_shapes_.size() == dtypes_.size(), in XRTAllocateFromTensorOp()
280 ctx, TensorShapeToXLAShape(dtypes_[i], tf_shapes_[i], &xla_shape)); in XRTAllocateFromTensorOp()
316 OP_REQUIRES(ctx, input_tensor.dtype() == dtypes_[i], in Compute()
357 DataTypeVector dtypes_; variable
538 OP_REQUIRES_OK(ctx, ctx->GetAttr("dtypes", &dtypes_)); in XRTReadToTensorOp()
616 DataTypeVector dtypes_; variable
|