Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_matrix.cc37 REGISTER_CSR_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
38 REGISTER_CSR_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
39 REGISTER_CSR_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_variant.cc81 REGISTER_RAGGED_TENSOR_VARIANT_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
82 REGISTER_RAGGED_TENSOR_VARIANT_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
84 VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
Dtensor_map.cc63 REGISTER_LIST_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
64 REGISTER_LIST_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
65 REGISTER_LIST_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
Dtensor_list.cc74 REGISTER_LIST_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
75 REGISTER_LIST_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
76 REGISTER_LIST_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry.cc119 const VariantDeviceCopyDirection direction, const Variant& from, in VariantDeviceCopy()
148 T, VariantDeviceCopyDirection::HOST_TO_DEVICE, \
151 T, VariantDeviceCopyDirection::DEVICE_TO_HOST, \
154 T, VariantDeviceCopyDirection::DEVICE_TO_DEVICE, \
Dvariant_op_registry.h56 enum VariantDeviceCopyDirection { enum
107 void RegisterDeviceCopyFn(const VariantDeviceCopyDirection direction, in RegisterDeviceCopyFn()
116 std::pair<std::pair<VariantDeviceCopyDirection, TypeIndex>, in RegisterDeviceCopyFn()
124 const VariantDeviceCopyDirection direction, const TypeIndex& type_index) { in GetDeviceCopyFn()
209 gtl::FlatMap<std::pair<VariantDeviceCopyDirection, TypeIndex>,
299 const VariantDeviceCopyDirection direction, const Variant& from,
391 const VariantDeviceCopyDirection direction, const TypeIndex& type_index, in UnaryVariantDeviceCopyRegistration()
Dvariant_op_registry_test.cc88 VariantValue, VariantDeviceCopyDirection::HOST_TO_DEVICE,
157 VariantDeviceCopyDirection::DEVICE_TO_DEVICE, in TEST()
162 VariantDeviceCopyDirection::HOST_TO_DEVICE, in TEST()
187 registry.RegisterDeviceCopyFn(VariantDeviceCopyDirection::HOST_TO_DEVICE, in TEST()
190 VariantDeviceCopyDirection::HOST_TO_DEVICE, kTypeIndex, f), in TEST()
Dvariant_op_copy_test.cc92 StoredTensorValue, VariantDeviceCopyDirection::HOST_TO_DEVICE,
96 StoredTensorValue, VariantDeviceCopyDirection::DEVICE_TO_HOST,
100 StoredTensorValue, VariantDeviceCopyDirection::DEVICE_TO_DEVICE,
Ddataset.cc202 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
203 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
204 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.cc101 VariantDeviceCopyDirection::HOST_TO_DEVICE, v[i], &v_out[i], copier); in CopyHostToDevice()
177 VariantDeviceCopy(VariantDeviceCopyDirection::DEVICE_TO_DEVICE, v[i], in CopyDeviceToDevice()
326 REGISTER_WRAPPED_TENSOR_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
327 REGISTER_WRAPPED_TENSOR_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
328 REGISTER_WRAPPED_TENSOR_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
380 VariantDeviceCopyDirection::DEVICE_TO_HOST, v[i], &v_out[i], copier); in CopyDeviceToHost()
/external/tensorflow/tensorflow/core/kernels/data/
Doptional_ops.cc60 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
61 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
62 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc725 s = VariantDeviceCopy(VariantDeviceCopyDirection::HOST_TO_DEVICE, in MakeTensorFromProto()