Searched refs:lhs_arg (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | matrix_ops_simple_test.cc | 211 auto lhs_arg = Parameter(&builder, 0, lhs_shape, "lhs"); in TestImpl() local
212 auto lhs_mat_arg = lhs_arg; in TestImpl()
220 result = Add(result, lhs_arg); in TestImpl()
|
| D | xla_hlo_profile_test.cc | 145 ScopedShapedBuffer lhs_arg, in ExecuteAndFetchProfile()
149 stream_ptr.get(), Literal::CreateFromShape(lhs_arg_shape), lhs_arg)); in ExecuteAndFetchProfile()
177 std::vector<const ShapedBuffer*> args = {&lhs_arg, &rhs_arg}; in ExecuteAndFetchProfile()
|
| D | dot_operation_test.cc | 826 auto lhs_arg = Parameter( in XLA_TYPED_TEST() local
835 lhs_arg = Transpose(lhs_arg, {1, 0}); in XLA_TYPED_TEST()
840 Dot(lhs_arg, rhs_arg); in XLA_TYPED_TEST()
|