Searched refs:platform_info_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 196 platform_info_(PlatformInfoFromContext(ctx)) {} in XlaLocalLaunchBase()
334 ctx, function_, platform_info_, resources_, constants_, /*lazy=*/false, in Compute()
336 if (!s.ok() && (platform_info_.device_type().type_string() == DEVICE_CPU || in Compute()
337 platform_info_.device_type().type_string() == DEVICE_GPU)) { in Compute()
352 client, platform_info_.allocator(), in Compute()
353 /*allocate_xla_tensors=*/platform_info_.is_on_xla_device(), in Compute()
354 platform_info_.UseMultipleStreams()); in Compute()
362 run_options.set_allocator(platform_info_.allocator()); in Compute()
438 platform_info_(PlatformInfoFromContext(ctx)), in XlaCompileOp()
460 ctx, function_, platform_info_, resources_, constants_, in Compute()
[all …]
|
| D | xla_ops.h | 121 const XlaPlatformInfo platform_info_; variable
156 XlaPlatformInfo platform_info_; variable
176 const XlaPlatformInfo platform_info_;
|