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

  /external/tensorflow/tensorflow/compiler/xla/service/
reshape_mover_test.cc 46 auto reshape0 = local
51 root_shape, HloOpcode::kAdd, reshape0, reshape1));
66 // +- reshape0 - rng0
72 // where rng0 has a different shape than reshape0.
82 auto reshape0 = local
89 root_shape, HloOpcode::kAdd, reshape0, const1));
109 auto reshape0 = local
114 root_shape, HloOpcode::kAdd, reshape0, reshape1));
135 auto reshape0 = local
140 root_shape, HloOpcode::kAdd, reshape0, reshape1))
223 auto reshape0 = local
259 auto reshape0 = local
313 auto reshape0 = local
352 auto reshape0 = local
380 auto reshape0 = local
412 auto reshape0 = local
536 auto reshape0 = local
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_instruction_fusion_test.cc 110 HloInstruction* reshape0 = local
114 MakeDot(ShapeUtil::MakeShape(F32, {1024, 1}), reshape0, arg1));

Completed in 251 milliseconds