Searched defs:dyn_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 1105 SmallVector<Value, 2> dyn_shape; in GetDotOpInitTensorDynSizes() local
1144 SmallVector<Value, 2> dyn_shape = GetDotOpInitTensorDynSizes( in matchAndRewrite() local
1181 SmallVector<Value, 8> dyn_shape; in GetDotGeneralOpInitTensorDynSizes() local
1230 SmallVector<Value, 8> dyn_shape = GetDotGeneralOpInitTensorDynSizes( in matchAndRewrite() local
1278 SmallVector<Value, 8> dyn_shape; in GetReduceOpInitTensorDynSizes() local
1350 SmallVector<Value, 8> dyn_shape = GetReduceOpInitTensorDynSizes( in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 81 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {4}); in AcceptDynamicR1() local
91 xla::Shape dyn_shape; in AcceptDynamicR2() local
118 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {4}); in AcceptDynamicR1Tuple() local
1139 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {4}); in TEST() local
1195 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {2, 4}); in TEST() local
1248 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {4}); in TEST() local
1314 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {4}); in TEST() local
1367 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {4}); in TEST() local
1411 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {2, 4}); in TEST() local
|