| /external/tensorflow/tensorflow/core/kernels/data/ |
| dataset_utils.cc | 31 if (!(return_values.size() == 1 && return_values[0].dtype() == DT_VARIANT &&
34 "Function must return a single scalar of dtype DT_VARIANT.");
|
| dense_to_sparse_batch_dataset_op.cc | 104 static DataTypeVector* output_dtypes_ = new DataTypeVector({DT_VARIANT});
258 Tensor serialized_sparse(DT_VARIANT, TensorShape({3}));
|
| group_by_window_dataset_op.cc | 457 Tensor group_dataset_arg(DT_VARIANT, TensorShape({}));
468 return_values[0].dtype() == DT_VARIANT &&
472 "DT_VARIANT.");
|
| tensor_queue_dataset_op.cc | 45 out.push_back(DT_VARIANT); // The queue component.
198 Tensor queues_t(cpu_allocator(), DT_VARIANT, TensorShape({batch_size}));
|
| /external/tensorflow/tensorflow/core/kernels/ |
| shape_op_test.cc | 77 auto input = ops::Placeholder(root, DT_VARIANT);
90 Tensor variant_tensor(DT_VARIANT, TensorShape({}));
103 Tensor variant_tensor(DT_VARIANT, TensorShape({1}));
111 Tensor variant_tensor(DT_VARIANT, TensorShape({}));
|
| shape_ops.h | 35 if (ctx->input_dtype(0) == DT_VARIANT) {
154 OP_REQUIRES(ctx, ctx->input(0).dtype() != DT_VARIANT,
207 OP_REQUIRES(ctx, ctx->input(0).dtype() != DT_VARIANT,
|
| resource_variable_ops.cc | 343 OP_REQUIRES(c, dtype_ == DT_VARIANT,
355 *ptr = new Var(DT_VARIANT);
359 OP_REQUIRES(context, variable->tensor()->dtype() == DT_VARIANT,
363 DataTypeString(DT_VARIANT)));
367 *variable->tensor() = Tensor(DT_VARIANT, value.shape());
|
| aggregate_ops.cc | 177 "AddN of non-scalar Tensor with dtype=DT_VARIANT is not "
199 Tensor out(cpu_allocator(), DT_VARIANT, TensorShape({}));
|
| serialize_sparse_op.cc | 107 *result = Tensor(DT_VARIANT, TensorShape({3}));
271 *result = Tensor(DT_VARIANT, TensorShape({n, 3}));
|
| constant_op.cc | 285 "dtype=DT_VARIANT is not supported."));
287 Tensor out(cpu_allocator(), DT_VARIANT, TensorShape({}));
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| copy_tensor.cc | 54 if (input->dtype() == DT_VARIANT) {
55 Tensor copy(cpu_allocator, DT_VARIANT, input->shape());
112 if (input->dtype() == DT_VARIANT) {
113 Tensor copy(cpu_allocator, DT_VARIANT, input->shape());
174 if (input->dtype() == DT_VARIANT) {
175 Tensor copy(cpu_allocator, DT_VARIANT, input->shape());
|
| rendezvous_mgr.cc | 81 if (!DataTypeCanUseMemcpy(in.dtype()) && in.dtype() != DT_VARIANT) {
104 if (in.dtype() != DT_VARIANT) {
|
| /external/tensorflow/tensorflow/core/framework/ |
| types.cc | 99 case DT_VARIANT:
195 *dt = DT_VARIANT;
|
| variant_op_copy_test.cc | 175 variant_proto.set_dtype(DT_VARIANT);
186 EXPECT_EQ(DT_VARIANT, outputs[0].dtype());
209 variant_proto.set_dtype(DT_VARIANT);
229 EXPECT_EQ(DT_VARIANT, outputs[0].dtype());
252 variant_proto.set_dtype(DT_VARIANT);
|
| tensor_util.cc | 42 CHECK_EQ(DT_VARIANT, other.dtype());
|
| dataset.cc | 24 // A wrapper class for storing a `DatasetBase` instance in a DT_VARIANT tensor.
29 // NOTE(mrry): This is not a feature-complete implementation of the DT_VARIANT
214 if (!(tensor.dtype() == DT_VARIANT ||
217 "Dataset tensor must be a scalar of dtype DT_VARIANT.");
232 if (!(tensor->dtype() == DT_VARIANT ||
235 "Dataset tensor must be a scalar of dtype DT_VARIANT.");
|
| variant_op_registry.cc | 59 CHECK_EQ(variant_tensor.dtype(), DT_VARIANT);
|
| types.h | 265 ToSet(DT_VARIANT) | ToSet(DT_UINT32) | ToSet(DT_UINT64) |
403 MATCH_TYPE_AND_ENUM(Variant, DT_VARIANT);
|
| dataset.h | 206 op_def->output_arg(0).type() == DT_VARIANT) ||
|
| /external/tensorflow/tensorflow/python/framework/ |
| dtypes.py | 124 types_pb2.DT_VARIANT, types_pb2.DT_VARIANT_REF, types_pb2.DT_RESOURCE,
302 if (self._type_enum == types_pb2.DT_VARIANT or
327 variant = DType(types_pb2.DT_VARIANT)
424 types_pb2.DT_VARIANT: variant,
474 types_pb2.DT_VARIANT: "variant",
|
| dtypes_test.py | 30 non_numeric_dtypes = [types_pb2.DT_VARIANT,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| base_rendezvous_mgr.cc | 248 if (!DMAHelper::CanUseDMA(&in) && in.dtype() != DT_VARIANT) {
273 if (in.dtype() != DT_VARIANT) {
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| tensor_bundle.cc | 206 DCHECK_NE(val.dtype(), DT_VARIANT);
269 DCHECK_EQ(val.dtype(), DT_VARIANT);
415 } else if (val.dtype() == DT_VARIANT) {
786 if (entry.dtype() != DT_STRING && entry.dtype() != DT_VARIANT) {
837 } else if (entry.dtype() == DT_VARIANT) {
[all...] |
| /external/tensorflow/tensorflow/c/eager/ |
| c_api.cc | 208 if (!dst_cpu && (src->dtype() != tensorflow::DT_VARIANT &&
219 if (src->dtype() == tensorflow::DT_VARIANT) {
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| gpu_device.cc | 595 if (parsed.dtype() == DT_VARIANT) {
597 Tensor copy(cpu_allocator(), DT_VARIANT, parsed.shape());
[all...] |
