Searched refs:use_tuple_arg (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.h | 109 bool use_tuple_arg = false; member
305 bool use_tuple_arg, xla::XlaBuilder* builder,
|
| D | xla_compiler.cc | 173 bool always_return_tuple, bool use_tuple_arg, bool alias_resource_update, in BuildComputation() argument
305 int param_num = use_tuple_arg ? 0 : update.input_index; in BuildComputation()
311 use_tuple_arg ? xla::ShapeIndex({update.input_index}) in BuildComputation()
820 options.use_tuple_arg, *options_.flib_def, debug_info, in CompileFunction()
955 bool use_tuple_arg, xla::XlaBuilder* builder, XlaContext* context, in BuildArguments() argument
1011 if (input_to_args->empty() && !use_tuple_arg) { in BuildArguments()
1038 if (use_tuple_arg) { in BuildArguments()
1053 if (use_tuple_arg) { in BuildArguments()
1334 *graph, real_args, options.use_tuple_arg, &builder, context, in CompileGraph()
1391 options.always_return_tuple, options.use_tuple_arg, in CompileGraph()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_scatter_ops.cc | 78 compile_options.use_tuple_arg = false; in Compile()
|
| D | reduce_window_op.cc | 83 compile_options.use_tuple_arg = false; in Compile()
|
| D | sort_ops.cc | 93 compile_options.use_tuple_arg = false; in Compile()
|
| D | xla_reduce_op.cc | 79 compile_options.use_tuple_arg = false; in Compile()
|
| D | xla_select_and_scatter_op.cc | 62 compile_options.use_tuple_arg = false; in Compile()
|
| D | while_op.cc | 337 body_options.use_tuple_arg = true; in Compile()
430 cond_options.use_tuple_arg = true; in Compile()
|
| D | case_op.cc | 156 options.use_tuple_arg = true; in Compile()
|
| D | if_op.cc | 243 options.use_tuple_arg = true; in Compile()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache.cc | 307 options.device_type.type_string(), compile_options.use_tuple_arg, in CompileSingleOp()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 382 compile_options.use_tuple_arg = true; in CompileTFFunctionToHlo()
|