HomeSort by relevance Sort by last modified time
    Searched refs:parameter_instruction (Results 1 - 22 of 22) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
tuple_util_test.cc 43 *param0 = (*entry_computation)->parameter_instruction(0);
44 *param1 = (*entry_computation)->parameter_instruction(1);
while_util.cc 25 narrow_condition->parameter_instruction(0)->shape();
43 TupleUtil::ExtractPrefix(wide_while_cond->parameter_instruction(0),
57 const Shape& narrow_shape = narrow_body->parameter_instruction(0)->shape();
67 HloInstruction* wide_parameter = wide_while_body->parameter_instruction(0);
122 HloInstruction* while_body_param = new_while_body->parameter_instruction(0);
while_util_test.cc 55 *param0 = (*entry_computation)->parameter_instruction(0);
56 *param1 = (*entry_computation)->parameter_instruction(1);
57 *param2 = (*entry_computation)->parameter_instruction(2);
source_map_util.cc 54 HloInstruction* param = computation.parameter_instruction(parameter_number);
defuser.cc 56 defused_instructions[fused_computation->parameter_instruction(i)] =
while_loop_simplifier.cc 139 auto* while_cond_param = while_cond->parameter_instruction(0);
164 auto* while_body_param = while_body->parameter_instruction(0);
201 // get-tuple-elem(comp->parameter_instruction(0), *indvar_tuple_idx),
318 for (const HloInstruction* instr : {while_body->parameter_instruction(0),
319 while_cond->parameter_instruction(0)}) {
343 HloInstruction* while_input = comp->parameter_instruction(0);
379 operand->operand(0) != while_body->parameter_instruction(0) ||
430 auto* param = comp->parameter_instruction(0);
while_loop_invariant_code_motion.cc 50 HloInstruction* while_body_param = while_body->parameter_instruction(0);
138 instr->operand(0) == while_body->parameter_instruction(0) &&
bfloat16_normalization_test.cc 200 ->parameter_instruction(0)
205 ->parameter_instruction(1)
hlo_dataflow_analysis.cc 636 add_to_worklist(user->true_computation()->parameter_instruction(0));
639 add_to_worklist(user->false_computation()->parameter_instruction(0));
648 called_computation->parameter_instruction(operand_number));
668 callsite.instruction()->while_body()->parameter_instruction(0));
670 callsite.instruction()->while_condition()->parameter_instruction(
    [all...]
hlo_computation.h 131 HloInstruction* parameter_instruction(int64 param_no) const { function in class:xla::HloComputation
reduce_precision_insertion.cc 132 ->parameter_instruction(i)));
liveness_util.cc 255 auto* param = user->to_apply()->parameter_instruction(operand_indices[0]);
buffer_assignment.cc     [all...]
hlo_verifier.cc 784 auto* cond_param = while_cond->parameter_instruction(0);
795 auto* body_param = while_body->parameter_instruction(0)
    [all...]
hlo_dataflow_analysis_test.cc     [all...]
copy_insertion.cc 257 HloInstruction* param = body->parameter_instruction(0);
    [all...]
layout_assignment.cc 576 condition->parameter_instruction(0)->shape())) {
    [all...]
hlo_instruction.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
while_transformer.cc 299 if (param0 != computation_->parameter_instruction(0)) {
313 computation_->parameter_instruction(0)) {
464 inst != computation_->parameter_instruction(0)) {
474 computation_->parameter_instruction(0)) {
ir_emitter_unnested.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
ir_emitter.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/tools/parser/
hlo_parser.cc 305 const Shape& param_shape = computation->parameter_instruction(p)->shape();
    [all...]

Completed in 675 milliseconds