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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_layout_assignment_test.cc 270 Shape dot_lhs_shape = ShapeUtil::MakeShape(F32, {m, k}); local
275 HloInstruction::CreateParameter(0, dot_lhs_shape, "param0"));
308 ShapeLayout(LayoutUtil::GetWithDefaultLayout(dot_lhs_shape));
cpu_instruction_fusion_test.cc 636 Shape dot_lhs_shape = ShapeUtil::MakeShape(F32, {m, k}); local
641 HloInstruction::CreateParameter(0, dot_lhs_shape, "param0"));
    [all...]

Completed in 248 milliseconds