Home
last modified time | relevance | path

Searched refs:VariantDeviceCopyDirection (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry.cc93 const VariantDeviceCopyDirection direction, const TypeIndex& type_index) { in GetDeviceCopyFn()
100 const VariantDeviceCopyDirection direction, const TypeIndex& type_index, in RegisterDeviceCopyFn()
108 std::pair<std::pair<VariantDeviceCopyDirection, TypeIndex>, in RegisterDeviceCopyFn()
114 const VariantDeviceCopyDirection direction, const Variant& from, in VariantDeviceCopy()
143 T, VariantDeviceCopyDirection::HOST_TO_DEVICE, \
146 T, VariantDeviceCopyDirection::DEVICE_TO_HOST, \
149 T, VariantDeviceCopyDirection::DEVICE_TO_DEVICE, \
Dvariant_op_registry_test.cc88 VariantValue, VariantDeviceCopyDirection::HOST_TO_DEVICE,
140 VariantDeviceCopyDirection::DEVICE_TO_DEVICE, in TEST()
145 VariantDeviceCopyDirection::HOST_TO_DEVICE, in TEST()
170 registry.RegisterDeviceCopyFn(VariantDeviceCopyDirection::HOST_TO_DEVICE, in TEST()
173 VariantDeviceCopyDirection::HOST_TO_DEVICE, kTypeIndex, f), in TEST()
Dvariant_op_registry.h52 enum VariantDeviceCopyDirection { enum
103 void RegisterDeviceCopyFn(const VariantDeviceCopyDirection direction,
110 const VariantDeviceCopyDirection direction, const TypeIndex& type_index);
162 gtl::FlatMap<std::pair<VariantDeviceCopyDirection, TypeIndex>,
252 const VariantDeviceCopyDirection direction, const Variant& from,
344 const VariantDeviceCopyDirection direction, const TypeIndex& type_index, in UnaryVariantDeviceCopyRegistration()
Dvariant_op_copy_test.cc92 StoredTensorValue, VariantDeviceCopyDirection::HOST_TO_DEVICE,
96 StoredTensorValue, VariantDeviceCopyDirection::DEVICE_TO_HOST,
100 StoredTensorValue, VariantDeviceCopyDirection::DEVICE_TO_DEVICE,
Ddataset.cc179 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
180 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
181 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.cc100 VariantDeviceCopyDirection::HOST_TO_DEVICE, v[i], &v_out[i], copier); in CopyHostToDevice()
165 VariantDeviceCopyDirection::DEVICE_TO_HOST, v[i], &v_out[i], copier); in CopyDeviceToHost()
229 VariantDeviceCopy(VariantDeviceCopyDirection::DEVICE_TO_DEVICE, v[i], in CopyDeviceToDevice()
382 REGISTER_WRAPPED_TENSOR_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
383 REGISTER_WRAPPED_TENSOR_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
384 REGISTER_WRAPPED_TENSOR_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
/external/tensorflow/tensorflow/core/kernels/data/
Doptional_ops.cc57 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
58 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
59 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.cc94 REGISTER_LIST_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
95 REGISTER_LIST_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
96 REGISTER_LIST_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc721 s = VariantDeviceCopy(VariantDeviceCopyDirection::HOST_TO_DEVICE, in MakeTensorFromProto()