Searched refs:param_0 (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_copy_insertion_test.cc | 70 auto param_0 = builder.AddInstruction( in TEST_F() local
89 HloInstruction::CreateWhile(scalar_shape_, condition, body, param_0)); in TEST_F()
|
| D | ir_emitter.cc | 1429 auto param_0 = function->parameter_instruction(0); in MatchReductionGenerator() local
1431 if (!(lhs == param_0 && rhs == param_1) && in MatchReductionGenerator()
1432 !(rhs == param_0 && lhs == param_1)) { in MatchReductionGenerator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | copy_insertion_test.cc | 1705 auto param_0 = builder.AddInstruction( in TEST_F() local
1754 {param_0, prev_element_1, prev_element_2, prev_element_3})); in TEST_F()
1795 auto param_0 = builder.AddInstruction( in TEST_F() local
1814 HloInstruction::CreateWhile(scalar_shape_, condition, body, param_0)); in TEST_F()
|
| D | hlo_parser_test.cc | 791 %fused_computation (constant.param_0: f32[3,2,1,1], constant.1.param_1: f32[2]) -> f32[3,2,1,1] { in CreateTestCases()
792 %constant.param_0 = f32[3,2,1,1]{3,2,1,0} parameter(0) in CreateTestCases()
795 …ROOT %subtract = f32[3,2,1,1]{3,2,1,0} subtract(f32[3,2,1,1]{3,2,1,0} %constant.param_0, f32[3,2,1… in CreateTestCases()
|