Home
last modified time | relevance | path

Searched defs:device_gpu (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/image/
Dresize_nearest_neighbor_op_test.cc44 std::unique_ptr<Device> device_gpu( in SetUp() local
83 std::unique_ptr<Device> device_gpu( in SetUp() local
Dresize_bilinear_op_test.cc46 std::unique_ptr<Device> device_gpu( in SetUp() local
/external/tensorflow/tensorflow/core/kernels/mlir_generated/
Dgpu_unary_ops_test.cc29 std::unique_ptr<tensorflow::Device> device_gpu( in SetUp() local
Dgpu_binary_ops_test.cc29 std::unique_ptr<tensorflow::Device> device_gpu( in SetUp() local