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 263 Shape dot_lhs_shape = ShapeUtil::MakeShape(F32, {m, k}); local
268 HloInstruction::CreateParameter(0, dot_lhs_shape, "param0"));
301 ShapeLayout(LayoutUtil::GetWithDefaultLayout(dot_lhs_shape));
cpu_instruction_fusion_test.cc 627 Shape dot_lhs_shape = ShapeUtil::MakeShape(F32, {m, k}); local
632 HloInstruction::CreateParameter(0, dot_lhs_shape, "param0"));

Completed in 562 milliseconds