Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dvariant_op_registry.cc129 const VariantDeviceCopyDirection direction, const Variant& from, in VariantDeviceCopy()
158 T, VariantDeviceCopyDirection::HOST_TO_DEVICE, \
161 T, VariantDeviceCopyDirection::DEVICE_TO_HOST, \
164 T, VariantDeviceCopyDirection::DEVICE_TO_DEVICE, \
H A Dvariant_op_registry.h56 enum VariantDeviceCopyDirection { enum
110 void RegisterDeviceCopyFn(const VariantDeviceCopyDirection direction, in RegisterDeviceCopyFn()
119 std::pair<std::pair<VariantDeviceCopyDirection, TypeIndex>, in RegisterDeviceCopyFn()
127 const VariantDeviceCopyDirection direction, const TypeIndex& type_index) { in GetDeviceCopyFn()
210 gtl::FlatMap<std::pair<VariantDeviceCopyDirection, TypeIndex>,
300 const VariantDeviceCopyDirection direction, const Variant& from,
392 const VariantDeviceCopyDirection direction, const TypeIndex& type_index, in UnaryVariantDeviceCopyRegistration()
H A 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()
H A Dvariant_op_copy_test.cc92 StoredTensorValue, VariantDeviceCopyDirection::HOST_TO_DEVICE,
96 StoredTensorValue, VariantDeviceCopyDirection::DEVICE_TO_HOST,
100 StoredTensorValue, VariantDeviceCopyDirection::DEVICE_TO_DEVICE,
H A Ddataset.cc219 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
220 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
221 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A 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);
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcopy_tensor.cc104 VariantDeviceCopyDirection::HOST_TO_DEVICE, v[i], &v_out[i], copier); in CopyHostToDevice()
179 VariantDeviceCopy(VariantDeviceCopyDirection::DEVICE_TO_DEVICE, v[i], in CopyDeviceToDevice()
334 REGISTER_WRAPPED_TENSOR_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
335 REGISTER_WRAPPED_TENSOR_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
336 REGISTER_WRAPPED_TENSOR_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
387 VariantDeviceCopyDirection::DEVICE_TO_HOST, v[i], &v_out[i], copier); in CopyDeviceToHost()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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);
H A 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);
H A Dlist_kernels.cc703 REGISTER_LIST_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
704 REGISTER_LIST_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
705 REGISTER_LIST_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A 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);
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A Dpluggable_device.cc399 s = VariantDeviceCopy(VariantDeviceCopyDirection::HOST_TO_DEVICE, in MakeTensorFromProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device.cc864 s = VariantDeviceCopy(VariantDeviceCopyDirection::HOST_TO_DEVICE, in MakeTensorFromProto()