Searched refs:lhs_arg (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ |
H A D | 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);
|
H A D | 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()},
|
H A D | 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 174 milliseconds