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

  /external/tensorflow/tensorflow/compiler/xla/tests/
matrix_ops_simple_test.cc 191 auto lhs_arg = builder.Parameter(0, lhs_shape, "lhs"); local
192 auto lhs_mat_arg = lhs_arg;
200 result = builder.Add(result, lhs_arg);
259 auto lhs_arg = builder.Parameter(0, lhs_shape, "lhs"); local
260 auto lhs_mat_arg = lhs_arg;
268 result = builder.Add(result, lhs_arg);
xla_hlo_profile_test.cc 131 std::unique_ptr<ScopedShapedBuffer> lhs_arg,
135 executor, *Literal::CreateFromShape(lhs_arg_shape), *lhs_arg));
167 executable->ExecuteOnStream(&run_options, {lhs_arg.get(), rhs_arg.get()},
dot_operation_test.cc 591 auto lhs_arg = builder.Parameter( local
598 lhs_arg = builder.Transpose(lhs_arg, {1, 0});
603 auto result = builder.Dot(lhs_arg, rhs_arg);

Completed in 91 milliseconds