Home
last modified time | relevance | path

Searched refs:is_entry_computation (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhuman_readable_profile_builder.h33 bool is_entry_computation, in HumanReadableProfileBuilder() argument
37 is_entry_computation_(is_entry_computation), in HumanReadableProfileBuilder()
Dhlo_parser.cc550 const bool is_entry_computation = EatIfPresent(TokKind::kw_ENTRY); in ParseComputation() local
581 if (is_entry_computation) { in ParseComputation()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.h205 bool is_entry_computation = true; member
359 Status XLAShapeForArgument(const Argument& arg, bool is_entry_computation,
438 bool is_entry_computation);
Dxla_compiler.cc613 bool is_entry_computation, in XLAShapeForArgument() argument
619 if (is_entry_computation) { in XLAShapeForArgument()
708 bool is_entry_computation) { in BuildArguments() argument
771 args[(*input_to_args)[i]], is_entry_computation, &arg_shapes[i])); in BuildArguments()
791 if (is_entry_computation) { in BuildArguments()
874 if (is_entry_computation) { in BuildArguments()
1093 options.is_entry_computation)); in CompileGraph()
1119 } else if (options.is_entry_computation) { in CompileGraph()
1164 options.is_entry_computation ? options_.shape_representation_fn in CompileGraph()
Dgraph_compiler.cc256 compile_options.is_entry_computation = false; in CompileFunctionalNode()
Dxla_compiler_test.cc1215 compile_options.is_entry_computation = false; // Only reshape variables. in TEST_F()
1285 compile_options.is_entry_computation = true; // Reshape args and retvals. in TEST_F()
1464 options.is_entry_computation = true; in TEST_F()
1482 options.is_entry_computation = false; in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_reduce_op.cc69 compile_options.is_entry_computation = false; in Compile()
Dreduce_window_op.cc85 compile_options.is_entry_computation = false; in Compile()
Dxla_select_and_scatter_op.cc64 compile_options.is_entry_computation = false; in Compile()
Dwhile_op.cc318 body_options.is_entry_computation = false; in Compile()
411 cond_options.is_entry_computation = false; in Compile()
Dif_op.cc101 options.is_entry_computation = false; in Compile()
Dcase_op.cc98 options.is_entry_computation = false; in Compile()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compile_on_demand_op.cc191 compile_options.is_entry_computation = true; in Compile()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc304 compile_options.is_entry_computation = true; in CompileToLocalExecutable()