Searched refs:rhs_arg (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | matrix_ops_simple_test.cc | 216 auto rhs_arg = Parameter(&builder, 1, rhs_shape, "rhs"); in TestImpl() local
217 auto result = Dot(lhs_mat_arg, rhs_arg); in TestImpl()
227 result = Add(result, rhs_arg); in TestImpl()
|
| D | xla_hlo_profile_test.cc | 152 ScopedShapedBuffer rhs_arg, in ExecuteAndFetchProfile()
156 stream_ptr.get(), Literal::CreateFromShape(rhs_arg_shape), rhs_arg)); in ExecuteAndFetchProfile()
177 std::vector<const ShapedBuffer*> args = {&lhs_arg, &rhs_arg}; in ExecuteAndFetchProfile()
|
| D | dot_operation_test.cc | 830 auto rhs_arg = Parameter( in XLA_TYPED_TEST() local
838 rhs_arg = Transpose(rhs_arg, {1, 0}); in XLA_TYPED_TEST()
840 Dot(lhs_arg, rhs_arg); in XLA_TYPED_TEST()
|