| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_v2_ops.cc | 146 OP_REQUIRES_OK(context, context->GetAttr("dtypes", &dtypes_)); in RestoreV2()
153 OP_REQUIRES(context, tensor_names.NumElements() == dtypes_.size(), in Compute()
155 " tensor names, but ", dtypes_.size(), in Compute()
184 shape_and_slices, dtypes_)); in Compute()
189 std::vector<DataType> dtypes_; member in tensorflow::RestoreV2
|
| D | resource_variable_ops.cc | 161 OP_REQUIRES_OK(c, c->GetAttr("dtypes", &dtypes_)); in ReadVariablesOp()
162 OP_REQUIRES(c, n == dtypes_.size(), in ReadVariablesOp()
165 " vs. ", dtypes_.size(), ")")); in ReadVariablesOp()
170 dtypes_.size()); in Compute()
171 std::vector<const ResourceHandle*> handles(dtypes_.size()); in Compute()
172 for (size_t i = 0; i < dtypes_.size(); ++i) { in Compute()
191 for (size_t i = 0; i < dtypes_.size(); ++i) { in Compute()
196 OP_REQUIRES(ctx, dtypes_[i] == variables[i]->tensor()->dtype(), in Compute()
200 " with wrong dtype. Expected ", DataTypeString(dtypes_[i]), in Compute()
|
| D | map_stage_op.cc | 95 DataTypeVector dtypes_ GUARDED_BY(mu_);
166 if (index >= dtypes_.size()) { in check_index()
169 "' was out of bounds '", dtypes_.size(), "'.")); in check_index()
214 "' was already initialized '", dtypes_.size(), "'.")); in check_index_uninitialized()
269 OptionalTuple empty(dtypes_.size()); in put_incomplete()
340 : dtypes_(dtypes), in StagingMap()
352 if (indices->NumElements() != dtypes_.size()) { in put()
|
| D | resource_variable_ops.h | 38 DataTypeVector dtypes_;
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | tensor_dataset_op.cc | 52 dtypes_.push_back(t.dtype()); in Dataset()
63 const DataTypeVector& output_dtypes() const override { return dtypes_; } in output_dtypes()
90 b->BuildAttrValue(dtypes_, &dtypes); in AsGraphDefInternal()
143 DataTypeVector dtypes_; member in tensorflow::data::__anon927a684b0111::TensorDatasetOp::Dataset
|
| D | tensor_slice_dataset_op.cc | 69 dtypes_.push_back(t.dtype()); in Dataset()
86 const DataTypeVector& output_dtypes() const override { return dtypes_; } in output_dtypes()
115 b->BuildAttrValue(dtypes_, &dtypes); in AsGraphDefInternal()
178 DataTypeVector dtypes_; member in tensorflow::data::__anon55e027fa0111::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()
213 const DataTypeVector dtypes_; member in tensorflow::data::__anone5092be70111::Dataset
|
| /external/tensorflow/tensorflow/contrib/ignite/kernels/dataset/ |
| D | ignite_dataset.cc | 43 dtypes_(dtypes), in IgniteDataset()
66 const DataTypeVector& IgniteDataset::output_dtypes() const { return dtypes_; } in output_dtypes()
|
| D | ignite_dataset.h | 57 const DataTypeVector dtypes_; variable
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 218 OP_REQUIRES_OK(ctx, ctx->GetAttr("dtypes", &dtypes_)); in XRTAllocateFromTensorOp()
225 ctx, tf_shapes_.size() == dtypes_.size(), in XRTAllocateFromTensorOp()
232 ctx, TensorShapeToXLAShape(dtypes_[i], tf_shapes_[i], &xla_shape)); in XRTAllocateFromTensorOp()
266 OP_REQUIRES(ctx, input_tensor.dtype() == dtypes_[i], in Compute()
310 DataTypeVector dtypes_; variable
512 OP_REQUIRES_OK(ctx, ctx->GetAttr("dtypes", &dtypes_)); in XRTReadToTensorOp()
592 DataTypeVector dtypes_; variable
|