Home
last modified time | relevance | path

Searched defs:device_spec (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dtpu_rewrite_device_util.cc122 auto lookup = [&devices](Device device_spec) { in GetTPUDevices()
145 for (const auto& device_spec : llvm::make_range( in GetTPUDevices() local
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dpywrap_tfe.h415 tensorflow::Safe_PyObjectPtr device_spec; member
H A Dcontext.py978 def device_spec(self): member in Context
982 def _set_device(self, device_name, device_spec): argument
H A Dpywrap_tfe_src.cc4100 tensorflow::Safe_PyObjectPtr device_spec; member
4126 PyObject* device_spec) {
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/
H A Dtflite_import_export.cc38 for (const auto& device_spec : device_specs) { in AttachCostPerDevice() local
/aosp_15_r20/external/arm-trusted-firmware/drivers/st/io/
H A Dio_mmc.c64 struct io_mmc_dev_spec *device_spec = in mmc_dev_open() local
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dnode_builder.cc105 NodeBuilder& NodeBuilder::Device(StringPiece device_spec) { in Device()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_helpers.cc171 DeviceNameUtils::ParsedName device_spec = device->parsed_name(); in GetTPUDevices() local
H A Dconfigure_tpu_embedding_rewrite_pass.cc228 DeviceNameUtils::ParsedName device_spec; in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/configuration/
H A Dproto_to_flatbuffer.cc271 FlatBufferBuilder* builder, const proto::EdgeTpuDeviceSpec& device_spec) { in ConvertEdgeTpuDeviceSpec()
H A Dflatbuffer_to_proto.cc281 const EdgeTpuDeviceSpec& device_spec) { in ConvertEdgeTpuDeviceSpec()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dnode_def_builder.cc209 NodeDefBuilder& NodeDefBuilder::Device(StringPiece device_spec) { in Device()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/hardwares/
H A Dtarget_hardware.cc271 for (auto& device_spec : specified_device_specs) { in ProcessTargetDevices() local
/aosp_15_r20/prebuilts/gcc/linux-x86/host/x86_64-linux-glibc2.17-4.8/sysroot/usr/include/X11/extensions/
DXKBstr.h423 unsigned short device_spec; member
519 unsigned short device_spec; member
580 unsigned short device_spec; member
/aosp_15_r20/hardware/google/gfxstream/host/apigen-codec-common/X11/extensions/
DXKBstr.h423 unsigned short device_spec; member
519 unsigned short device_spec; member
580 unsigned short device_spec; member
/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dtfe_wrapper.cc489 py::handle device_spec) in EagerContextThreadLocalDataWrapper()