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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
stream_assignment_test.cc 54 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, x, y));
56 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, dot1, z));
73 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, x, y));
75 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, y, x));
105 CreateCanonicalDot(f32_2x2_, params[2], params[3]));
107 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, params[1], d00));
109 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, d00, params[4]));
111 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, params[0], d10));
113 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, d10, d11));
115 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, d11, params[5]))
    [all...]
gpu_hlo_schedule_test.cc 78 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, x, y));
80 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, dot1, z));
206 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, x, y));
208 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, y, x));
210 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, dot1, dot2));
273 CreateCanonicalDot(f32_2x2_, params[2], params[3]));
275 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, params[1], d00));
277 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, d00, params[4]));
279 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, params[0], d10));
281 builder.AddInstruction(CreateCanonicalDot(f32_2x2_, d10, d11))
    [all...]
instruction_fusion_test.cc 116 CreateCanonicalDot(ShapeUtil::MakeShape(S32, {1, 1}), param0, param0));
133 CreateCanonicalDot(ShapeUtil::MakeShape(S32, {1, 1}), param0, param0));
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
cpu_eigen_dot_operation_test.cc 72 builder.AddInstruction(CreateCanonicalDot(param_shape, lhs, rhs));
89 builder.AddInstruction(CreateCanonicalDot(param_shape, lhs_transposed, rhs));
  /external/tensorflow/tensorflow/compiler/xla/tests/
test_utils.h 104 std::unique_ptr<HloDotInstruction> CreateCanonicalDot(const Shape& shape,
test_utils.cc 575 std::unique_ptr<HloDotInstruction> CreateCanonicalDot(const Shape& shape,
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_layout_assignment_test.cc 74 CreateCanonicalDot(result_shape, dot_lhs, dot_rhs));
111 CreateCanonicalDot(result_shape, dot_a_lhs, dot_rhs));
113 CreateCanonicalDot(result_shape, dot_b_lhs, dot_rhs));
155 CreateCanonicalDot(result_a_shape, dot_a_lhs, dot_rhs));
157 CreateCanonicalDot(result_b_shape, dot_b_lhs, dot_rhs));
193 CreateCanonicalDot(result_shape, dot_lhs, dot_rhs));
233 CreateCanonicalDot(result_shape, dot_lhs, dot_rhs));
281 builder.AddInstruction(CreateCanonicalDot(dot_shape, dot_lhs, dot_rhs));
cpu_instruction_fusion_test.cc 648 builder.AddInstruction(CreateCanonicalDot(dot_shape, dot_lhs, dot_rhs));
    [all...]

Completed in 269 milliseconds