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

  /external/tensorflow/tensorflow/compiler/xla/service/
dot_decomposer.cc 39 const Shape& dot_shape = dot->shape(); local
86 PrimitiveType dot_type = dot_shape.element_type();
87 const int64 dot_rows = dot_shape.dimensions(num_batch_dims + 0);
88 const int64 dot_cols = dot_shape.dimensions(num_batch_dims + 1);
152 HloInstruction::CreateReshape(dot_shape, concat));
algebraic_simplifier_test.cc 2582 Shape dot_shape = ShapeUtil::MakeShape(F32, {m, n}); local
2679 Shape dot_shape = ShapeUtil::MakeShape(F32, {spec.m, spec.n}); local
2743 Shape dot_shape = ShapeUtil::MakeShape(F32, {spec.m, spec.n}); local
    [all...]
algebraic_simplifier.cc 284 const Shape& dot_shape, HloInstruction* lhs, int64 lhs_contracting_dim,
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_layout_assignment_test.cc 265 Shape dot_shape = ShapeUtil::MakeShape(F32, {m, n}); local
270 HloInstruction::CreateParameter(1, dot_shape, "param1"));
274 HloInstruction::CreateCanonicalDot(dot_shape, dot_lhs, dot_rhs));
278 dot_shape, HloOpcode::kAdd, dot_result, addend));
281 dot_shape, HloOpcode::kAdd, addend, dot_result));
288 dot_shape, HloInstruction::FusionKind::kOutput, add_result));
303 ShapeLayout(LayoutUtil::GetWithDefaultLayout(dot_shape));
305 ShapeLayout(LayoutUtil::GetWithDefaultLayout(dot_shape));
cpu_instruction_fusion_test.cc 629 Shape dot_shape = ShapeUtil::MakeShape(F32, {m, n}); local
636 HloInstruction::CreateParameter(2, dot_shape, "param2"));
639 HloInstruction::CreateCanonicalDot(dot_shape, dot_lhs, dot_rhs));
641 HloInstruction::CreateBinary(dot_shape, HloOpcode::kAdd, dot, addend));
645 HloInstruction::CreateOutfeed(dot_shape, dot, "no_config"));

Completed in 401 milliseconds