/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
cpu_copy_insertion_test.cc | 70 auto param_0 = builder.AddInstruction( local 71 HloInstruction::CreateParameter(0, scalar_shape_, "param_0")); 89 HloInstruction::CreateWhile(scalar_shape_, condition, body, param_0)); 109 HloInstruction::CreateParameter(0, scalar_shape_, "param_0"));
|
ir_emitter.cc | [all...] |
/external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
hlo_parser_test.cc | 676 %fused_computation (constant.param_0: f32[3,2,1,1], constant.1.param_1: f32[2]) -> f32[3,2,1,1] { 677 %constant.param_0 = f32[3,2,1,1]{3,2,1,0} parameter(0) 680 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,1]{3,2,1,0} %broadcast) [all...] |
/external/tensorflow/tensorflow/compiler/xla/tests/ |
map_test.cc | 348 auto param_0 = local 351 auto adder_to_one = embedded_builder.Add(param_0, constant_one);
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
copy_insertion_test.cc | 1476 auto param_0 = builder.AddInstruction( local 1566 auto param_0 = builder.AddInstruction( local [all...] |