| /external/tensorflow/tensorflow/compiler/xla/service/ |
| reshape_mover_test.cc | 48 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param0));
50 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param1));
89 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, rng0));
117 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param0));
119 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param1));
144 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param0));
146 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param1));
194 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param1));
199 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param2));
235 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param0))
[all...] |
| dot_decomposer.cc | 98 HloInstruction::CreateReshape(lhs_shape_r3, lhs));
100 HloInstruction::CreateReshape(rhs_shape_r3, rhs));
110 HloInstruction::CreateReshape(lhs_slice_shape_r2, lhs_slice_r3));
117 HloInstruction::CreateReshape(rhs_slice_shape_r2, rhs_slice_r3));
145 HloInstruction::CreateReshape(dot_shape_r3, dot_r2));
155 HloInstruction::CreateReshape(dot_shape, concat));
216 computation->AddInstruction(HloInstruction::CreateReshape(
261 computation->AddInstruction(HloInstruction::CreateReshape(
282 original_dot, computation->AddInstruction(HloInstruction::CreateReshape(
|
| dynamic_index_splitter.cc | 78 auto bitcast = parent->AddInstruction(HloInstruction::CreateReshape(
|
| convolution_group_converter.cc | 361 add(HloInstruction::CreateReshape(reshaped_filter_shape, filter));
414 HloInstruction::CreateReshape(reshaped_activation_shape, activation));
421 add(HloInstruction::CreateReshape(reshaped_filter_shape, filter));
450 auto reshaped_convolution = HloInstruction::CreateReshape(
|
| reshape_mover.cc | 150 HloInstruction::CreateReshape(new_shape, operand));
244 HloInstruction::CreateReshape(instruction->shape(), new_elementwise);
|
| bfloat16_propagation_test.cc | 579 builder_cond.AddInstruction(HloInstruction::CreateReshape(
584 builder_cond.AddInstruction(HloInstruction::CreateReshape(
637 builder_cond.AddInstruction(HloInstruction::CreateReshape(
642 builder_cond.AddInstruction(HloInstruction::CreateReshape(
712 builder_cond.AddInstruction(HloInstruction::CreateReshape(
717 builder_cond.AddInstruction(HloInstruction::CreateReshape(
[all...] |
| hlo_creation_utils.cc | 111 HloInstruction::CreateReshape(result_shape, operand));
134 computation->AddInstruction(HloInstruction::CreateReshape(
162 computation->AddInstruction(HloInstruction::CreateReshape(
|
| instruction_fusion_test.cc | 118 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {1, 1}), param0));
134 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {1, 1}), param0));
|
| algebraic_simplifier_test.cc | [all...] |
| hlo_rematerialization_test.cc | 64 HloInstruction::CreateReshape(scalar_shape_, param));
109 HloInstruction::CreateReshape(scalar_shape_, param));
|
| algebraic_simplifier.cc | 268 return computation_->AddInstruction(HloInstruction::CreateReshape(
277 return computation_->AddInstruction(HloInstruction::CreateReshape(
[all...] |
| dynamic_dimension_inference_test.cc | 386 HloInstruction::CreateReshape(output_shape, a_param));
420 builder.AddInstruction(HloInstruction::CreateReshape(output_shape, a_param));
|
| while_loop_simplifier.cc | [all...] |
| layout_assignment_test.cc | 359 builder.AddInstruction(HloInstruction::CreateReshape(bshape, log));
556 HloInstruction::CreateReshape(cshape, concatenate));
644 builder.AddInstruction(HloInstruction::CreateReshape(input_shape, param));
[all...] |
| pattern_matcher_test.cc | [all...] |
| hlo_instruction.h | 687 static std::unique_ptr<HloInstruction> CreateReshape(const Shape& shape,
[all...] |
| hlo_instruction.cc | [all...] |
| heap_simulator_test.cc | 144 cond_builder.AddInstruction(HloInstruction::CreateReshape(r0f32, slice));
[all...] |
| hlo_parser.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| fusion_test.cc | 336 auto reshape = builder.AddInstruction(HloInstruction::CreateReshape(
351 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape(
367 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {3, 2}), const0));
382 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {}), const0));
396 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape(
412 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {}), const0));
427 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {3, 3}), const0));
564 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {2, 2}), const0));
[all...] |
| multioutput_fusion_test.cc | 149 builder.AddInstruction(HloInstruction::CreateReshape(
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_instruction_fusion_test.cc | 117 builder.AddInstruction(HloInstruction::CreateReshape(
320 builder.AddInstruction(HloInstruction::CreateReshape(result_shape, exp1));
348 HloInstruction::CreateReshape(reshape_shape, broadcast2));
431 HloInstruction::CreateReshape(result_shape, param0));
597 auto idx_choice = builder.AddInstruction(HloInstruction::CreateReshape(
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| instruction_fusion_test.cc | 81 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {}), exp1));
117 auto reshape2 = builder.AddInstruction(HloInstruction::CreateReshape(
|
| cudnn_conv_rewriter.cc | 487 rhs = c->AddInstruction(HloInstruction::CreateReshape(new_shape, rhs));
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| cpu_fusion_test.cc | 171 builder.AddInstruction(HloInstruction::CreateReshape(
|
