Searched refs:MatMulShape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime_test.cc | 116 struct MatMulShape { struct
122 MatMulShape MatMulShapes[] = {
123 MatMulShape{2, 2, 3}, MatMulShape{256, 512, 1024},
124 MatMulShape{128, 128, 1}, MatMulShape{1, 128, 128},
125 MatMulShape{1, 32, 128}, MatMulShape{1, 32, 16},
126 MatMulShape{32, 16, 1}, MatMulShape{32, 128, 1},
134 using MatMulTestParam = std::tuple<MatMulShape, bool, bool, bool>;
140 MatMulShape shape = std::get<0>(info.param); in Name()
153 MatMulShape shape = std::get<0>(GetParam()); in TEST_P()
177 MatMulShape shape = std::get<0>(info.param); in Name()
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns_test.cc | 122 TF_EXPECT_OK(MatMulShape(&c)); in TEST()
132 TF_EXPECT_OK(MatMulShape(&c)); in TEST()
142 auto s = MatMulShape(&c); in TEST()
152 TF_EXPECT_OK(MatMulShape(&c)); in TEST()
162 auto s = MatMulShape(&c); in TEST()
173 auto s = MatMulShape(&c); in TEST()
191 auto s = MatMulShape(&c); in TEST()
209 auto s = MatMulShape(&c); in TEST()
|
| D | common_shape_fns.h | 227 Status MatMulShape(shape_inference::InferenceContext* c);
|
| D | common_shape_fns.cc | 213 Status MatMulShape(shape_inference::InferenceContext* c) { in MatMulShape() function
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops.cc | 834 .SetShapeFn(shape_inference::MatMulShape);
846 .SetShapeFn(shape_inference::MatMulShape);
1580 TF_RETURN_IF_ERROR(shape_inference::MatMulShape(c)); in __anonb22bfa860f02()
|