Searched refs:is_entry_computation (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | human_readable_profile_builder.h | 33 bool is_entry_computation, in HumanReadableProfileBuilder() argument
37 is_entry_computation_(is_entry_computation), in HumanReadableProfileBuilder()
|
| D | hlo_parser.cc | 873 const bool is_entry_computation = EatIfPresent(TokKind::kw_ENTRY); in ParseComputation() local
904 if (is_entry_computation) { in ParseComputation()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.h | 125 bool is_entry_computation = true; member
230 const Argument& arg, bool is_entry_computation,
311 bool is_entry_computation);
|
| D | xla_compiler.cc | 172 bool is_entry_computation, bool return_updated_values_for_all_resources, in BuildComputation() argument
306 if (is_entry_computation && in BuildComputation()
360 if (retval_index_and_sharding.empty() || !is_entry_computation) { in BuildComputation()
834 const XlaCompiler::Argument& arg, bool is_entry_computation, in XLAShapeForArgument() argument
841 if (is_entry_computation) { in XLAShapeForArgument()
959 bool is_entry_computation) { in BuildArguments() argument
1034 is_entry_computation, sharding, in BuildArguments()
1055 if (is_entry_computation) { in BuildArguments()
1099 if (is_entry_computation) { in BuildArguments()
1157 if (is_entry_computation) { in BuildArguments()
[all …]
|
| D | graph_compiler.cc | 263 compile_options.is_entry_computation = false; in CompileFunctionalNode()
|
| D | xla_compiler_test.cc | 1283 compile_options.is_entry_computation = false; // Only reshape variables. in TEST_F()
1354 compile_options.is_entry_computation = true; // Reshape args and retvals. in TEST_F()
1532 options.is_entry_computation = true; in TEST_F()
1550 options.is_entry_computation = false; in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_scatter_ops.cc | 80 compile_options.is_entry_computation = false; in Compile()
|
| D | reduce_window_op.cc | 84 compile_options.is_entry_computation = false; in Compile()
|
| D | sort_ops.cc | 95 compile_options.is_entry_computation = false; in Compile()
|
| D | xla_reduce_op.cc | 81 compile_options.is_entry_computation = false; in Compile()
|
| D | xla_select_and_scatter_op.cc | 63 compile_options.is_entry_computation = false; in Compile()
|
| D | while_op.cc | 339 body_options.is_entry_computation = false; in Compile()
431 cond_options.is_entry_computation = false; in Compile()
|
| D | case_op.cc | 158 options.is_entry_computation = false; in Compile()
|
| D | if_op.cc | 245 options.is_entry_computation = false; in Compile()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 138 compile_options.is_entry_computation = true; in Compile()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 200 compile_options.is_entry_computation = true; in CompileToLocalExecutable()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 383 compile_options.is_entry_computation = true; in CompileTFFunctionToHlo()
|