Home
last modified time | relevance | path

Searched refs:MatMulShape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime_test.cc114 struct MatMulShape { struct
120 MatMulShape MatMulShapes[] = {
121 MatMulShape{2, 2, 3}, MatMulShape{256, 512, 1024},
122 MatMulShape{128, 128, 1}, MatMulShape{1, 128, 128},
123 MatMulShape{1, 32, 128}, MatMulShape{1, 32, 16},
124 MatMulShape{32, 16, 1}, MatMulShape{32, 128, 1},
132 using MatMulTestParam = std::tuple<MatMulShape, bool, bool, bool>;
138 MatMulShape shape = std::get<0>(info.param); in Name()
151 MatMulShape shape = std::get<0>(GetParam()); in TEST_P()
175 MatMulShape shape = std::get<0>(info.param); in Name()
[all …]
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns_test.cc122 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()
Dcommon_shape_fns.h107 Status MatMulShape(shape_inference::InferenceContext* c);
Dcommon_shape_fns.cc110 Status MatMulShape(shape_inference::InferenceContext* c) { in MatMulShape() function
/external/tensorflow/tensorflow/core/ops/
Dmkl_nn_ops.cc304 .SetShapeFn(shape_inference::MatMulShape)
329 .SetShapeFn(shape_inference::MatMulShape)
1324 TF_RETURN_IF_ERROR(shape_inference::MatMulShape(c)); in __anon8bfbd4921402()
1368 TF_RETURN_IF_ERROR(shape_inference::MatMulShape(c)); in __anon8bfbd4921502()
1416 TF_RETURN_IF_ERROR(shape_inference::MatMulShape(c)); in __anon8bfbd4921602()
1461 TF_RETURN_IF_ERROR(shape_inference::MatMulShape(c)); in __anon8bfbd4921702()
1509 TF_RETURN_IF_ERROR(shape_inference::MatMulShape(c)); in __anon8bfbd4921802()
Dmath_ops.cc922 .SetShapeFn(shape_inference::MatMulShape);
932 .SetShapeFn(shape_inference::MatMulShape);
945 .SetShapeFn(shape_inference::MatMulShape);
962 .SetShapeFn(shape_inference::MatMulShape)
1826 TF_RETURN_IF_ERROR(shape_inference::MatMulShape(c)); in __anon57f5a5dd1202()
Drisc_ops.cc169 .SetShapeFn(shape_inference::MatMulShape);
Dnn_ops.cc3053 TF_RETURN_IF_ERROR(shape_inference::MatMulShape(c)); in __anon4650176f4002()
3083 TF_RETURN_IF_ERROR(shape_inference::MatMulShape(c)); in __anon4650176f4102()
3116 TF_RETURN_IF_ERROR(shape_inference::MatMulShape(c)); in __anon4650176f4202()
3149 TF_RETURN_IF_ERROR(shape_inference::MatMulShape(c)); in __anon4650176f4302()
3183 TF_RETURN_IF_ERROR(shape_inference::MatMulShape(c)); in __anon4650176f4402()