experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Initialize ExecutionOptions in ComputeConstant to default values.

Browse Source 
PiperOrigin-RevId: 164894867
This commit is contained in:
A. Unique TensorFlower 2017-08-10 13:28:30 -07:00 committed by TensorFlower Gardener
parent c8897e9bce
commit 7dfabcc01c
2 changed files with 5 additions and 1 deletions
tensorflow/compiler/xla/service
BUILDservice.cc

1
tensorflow/compiler/xla/service/BUILD
View File

@ -436,6 +436,7 @@ cc_library(
":user_computation", ":user_computation",
":versioned_computation_handle", ":versioned_computation_handle",
"//tensorflow/compiler/xla:executable_run_options", "//tensorflow/compiler/xla:executable_run_options",
"//tensorflow/compiler/xla:execution_options_util",
"//tensorflow/compiler/xla:service_interface", "//tensorflow/compiler/xla:service_interface",
"//tensorflow/compiler/xla:shape_layout", "//tensorflow/compiler/xla:shape_layout",
"//tensorflow/compiler/xla:shape_util", "//tensorflow/compiler/xla:shape_util",

5
tensorflow/compiler/xla/service/service.cc
View File

@ -20,6 +20,7 @@ limitations under the License.
#include <utility> #include <utility>
#include <vector> #include <vector>
#include "tensorflow/compiler/xla/execution_options_util.h"
#include "tensorflow/compiler/xla/layout_util.h" #include "tensorflow/compiler/xla/layout_util.h"
#include "tensorflow/compiler/xla/legacy_flags/debug_options_flags.h" #include "tensorflow/compiler/xla/legacy_flags/debug_options_flags.h"
#include "tensorflow/compiler/xla/ptr_util.h" #include "tensorflow/compiler/xla/ptr_util.h"
@ -1111,8 +1112,10 @@ tensorflow::Status Service::ComputeConstant(const ComputeConstantRequest* arg,
TF_DCHECK_OK(ShapeUtil::ValidateShape(program_shape.result())); TF_DCHECK_OK(ShapeUtil::ValidateShape(program_shape.result()));
ExecutionOptions execution_options; ExecutionOptions execution_options = xla::CreateDefaultExecutionOptions();
execution_options.mutable_debug_options()->set_xla_enable_fast_math(false); execution_options.mutable_debug_options()->set_xla_enable_fast_math(false);
execution_options.mutable_debug_options()->set_xla_backend_optimization_level(
0);
*execution_options.mutable_shape_with_output_layout() = *execution_options.mutable_shape_with_output_layout() =
program_shape.result(); program_shape.result();