Home
last modified time | relevance | path

Searched refs:XlaPlatformInfoFromDevice (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_compile_on_demand_op.h41 platform_info_(XlaPlatformInfoFromDevice(ctx->device())) {} in XlaCompileOnDemandOp()
Dxla_platform_info.h88 XlaPlatformInfo XlaPlatformInfoFromDevice(DeviceBase* device);
Dxla_platform_info.cc78 XlaPlatformInfo XlaPlatformInfoFromDevice(DeviceBase* device_base) { in XlaPlatformInfoFromDevice() function
Dget_compiler_ir.cc93 XlaPlatformInfo platform_info = XlaPlatformInfoFromDevice(dev); in GetCompilerIr()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc161 platform_info_(XlaPlatformInfoFromDevice(ctx->device())), in XlaLocalLaunchBase()
366 platform_info_(XlaPlatformInfoFromDevice(ctx->device())), in XlaCompileOp()
456 : OpKernel(ctx), platform_info_(XlaPlatformInfoFromDevice(ctx->device())) {} in XlaRunOp()