Home
last modified time | relevance | path

Searched defs:default_device (Results 1 – 25 of 25) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtf_device_assignment.cc35 explicit SimpleTFDeviceAssignmentPass(llvm::StringRef default_device) { in SimpleTFDeviceAssignmentPass()
98 llvm::StringRef default_device) { in CreateSimpleTFDeviceAssignmentPass()
/aosp_15_r20/external/vulkan-validation-layers/tests/
H A Dtest_environment.h43 Device &default_device() { return *(devs_[default_dev_]); } in default_device() function
/aosp_15_r20/external/libdrm/tests/tegra/
H A Dopenclose.c30 static const char default_device[] = "/dev/dri/card0"; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/runtime/
H A Druntime.cc48 std::string default_device = kDefaultHostDeviceName; in InitializeOpHandlers() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dgraph_constructor.h155 string default_device; member
H A Dinspecting_placer.cc116 const Device* default_device, in InspectingPlacer()
H A Dprocess_function_library_runtime.cc462 const FunctionLibraryDefinition* lib_def, Device* default_device) { in PinArgsAndRets()
817 Device* default_device = nullptr; in InstantiateMultiDevice() local
H A Dgraph_constructor.cc153 string default_device; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/translate/
H A Dtfrt_compile_options.h39 std::string default_device = "/job:localhost/replica:0/task:0/device:CPU:0"; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Djax_jit.h58 std::optional<pybind11::object> default_device; member
H A Djax_jit.cc686 xla::PjRtDevice* default_device, bool is_committed, in ComputeSignature()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dtransform_graph_function.cc83 Device* default_device = nullptr; in TransformGraphFunction() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/
H A Dsoftmax_op_benchmark.cc83 Eigen::DefaultDevice default_device; in EigenSoftmax() local
/aosp_15_r20/external/nos/host/generic/libnos_datagram/
Dcitadel.cpp337 static const char *default_device(void) { in default_device() function
/aosp_15_r20/external/pdfium/core/fxge/skia/
H A Dfx_skia_device_embeddertest.cpp180 auto default_device = std::make_unique<CFX_DefaultRenderDevice>(); in RenderPageToSkCanvas() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dshape_analysis.cpp561 at::Device default_device = at::kCPU; in propagateTorchTensorShape() local
1746 c10::Device default_device = *type->device(); in PropagateTensorShapeOnNode() local
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/backends/gpu/
H A Drocm_tracer.cc319 thread_local uint32_t default_device = 0; in operator ()() local
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_executor.cpp654 Eigen::DefaultDevice default_device; in EIGEN_DECLARE_TEST() local
/aosp_15_r20/external/webrtc/modules/audio_device/win/
H A Dcore_audio_utility_win.cc552 ComPtr<IMMDevice> default_device; in GetDeviceNamesInternal() local
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_arg_parser.h840 const at::Device& default_device) { in deviceWithDefault()
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistribute_lib.py377 default_device=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
H A Dcontrol_flow_ops_py_test.py4992 default_device, argument
/aosp_15_r20/external/ComputeLibrary/include/CL/
H A Dopencl.hpp2217 static Device setDefault(const Device &default_device) in setDefault()
/aosp_15_r20/external/mesa3d/include/CL/
H A Dopencl.hpp2211 static Device setDefault(const Device &default_device) in setDefault()
/aosp_15_r20/external/OpenCL-CLHPP/include/CL/
H A Dopencl.hpp2499 static Device setDefault(const Device &default_device) in setDefault()