HomeSort by relevance Sort by last modified time
    Searched refs:VariantDeviceCopyDirection (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/core/framework/
variant_op_registry.cc 146 const VariantDeviceCopyDirection direction, StringPiece type_name) {
153 const VariantDeviceCopyDirection direction, const string& type_name,
161 std::pair<std::pair<VariantDeviceCopyDirection, StringPiece>,
168 const VariantDeviceCopyDirection direction, const Variant& from,
variant_op_registry.h 48 enum VariantDeviceCopyDirection {
106 void RegisterDeviceCopyFn(const VariantDeviceCopyDirection direction,
113 const VariantDeviceCopyDirection direction, StringPiece type_name);
164 std::unordered_map<std::pair<VariantDeviceCopyDirection, StringPiece>,
262 const VariantDeviceCopyDirection direction, const Variant& from,
375 const VariantDeviceCopyDirection direction, const string& type_name,
495 // VariantDeviceCopyDirection, and the device_copy_fn has signature:
variant_op_registry_test.cc 97 VariantValue, VariantDeviceCopyDirection::HOST_TO_DEVICE,
184 VariantDeviceCopyDirection::DEVICE_TO_DEVICE, "TEST VariantValue"),
188 VariantDeviceCopyDirection::HOST_TO_DEVICE, "TEST VariantValue");
211 registry.RegisterDeviceCopyFn(VariantDeviceCopyDirection::HOST_TO_DEVICE,
214 VariantDeviceCopyDirection::HOST_TO_DEVICE, kTypeName, f),
variant_op_copy_test.cc 91 StoredTensorValue, VariantDeviceCopyDirection::HOST_TO_DEVICE,
95 StoredTensorValue, VariantDeviceCopyDirection::DEVICE_TO_HOST,
99 StoredTensorValue, VariantDeviceCopyDirection::DEVICE_TO_DEVICE,
  /external/tensorflow/tensorflow/core/common_runtime/
copy_tensor.cc 93 VariantDeviceCopyDirection::HOST_TO_DEVICE, v[i], &v_out[i], copier);
151 VariantDeviceCopyDirection::DEVICE_TO_HOST, v[i], &v_out[i], copier);
215 VariantDeviceCopy(VariantDeviceCopyDirection::DEVICE_TO_DEVICE, v[i],
  /external/tensorflow/tensorflow/core/kernels/
list_kernels.cc 84 REGISTER_LIST_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
85 REGISTER_LIST_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
86 REGISTER_LIST_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
resource_variable_ops.cc 374 VariantDeviceCopyDirection::DEVICE_TO_DEVICE,
  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device.cc 618 s = VariantDeviceCopy(VariantDeviceCopyDirection::HOST_TO_DEVICE,
    [all...]

Completed in 395 milliseconds