Searched refs:platform_info_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 55 platform_info_); in Run()
58 /*allocate_xla_tensors=*/platform_info_.xla_device_metadata() != nullptr, in Run()
59 platform_info_.xla_device_metadata() in Run()
60 ? platform_info_.xla_device_metadata()->UseMultipleStreams() in Run()
125 return BuildXlaCompilationCache(ctx->device(), platform_info_, in Compile()
133 platform_info_, in Compile()
|
| D | xla_compile_on_demand_op.h | 41 platform_info_(XlaPlatformInfoFromDevice(ctx->device())) {} in XlaCompileOnDemandOp()
57 const XlaPlatformInfo platform_info_; variable
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 161 platform_info_(XlaPlatformInfoFromDevice(ctx->device())), in XlaLocalLaunchBase()
235 ctx, function_, /*has_ref_vars=*/has_ref_vars_, platform_info_, inputs, in Compute()
254 platform_info_); in Compute()
259 /*allocate_xla_tensors=*/platform_info_.is_on_xla_device(), in Compute()
260 platform_info_.UseMultipleStreams()); in Compute()
280 if (!stream || platform_info_.platform_id() == se::host::kHostPlatformId) { in Compute()
366 platform_info_(XlaPlatformInfoFromDevice(ctx->device())), in XlaCompileOp()
398 ctx, function_, has_ref_vars_, platform_info_, inputs, variable_infos, in Compute()
456 : OpKernel(ctx), platform_info_(XlaPlatformInfoFromDevice(ctx->device())) {} in XlaRunOp()
470 platform_info_); in Compute()
[all …]
|
| D | xla_ops.h | 66 const XlaPlatformInfo platform_info_; variable
103 XlaPlatformInfo platform_info_; variable
127 const XlaPlatformInfo platform_info_;
|