Searched refs:dyn_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | slot_creator_test.py | 82 dyn_shape = constant_op.constant([2], dtype=dtypes.int32)
83 dyn_shape = array_ops.placeholder_with_default(dyn_shape,
87 initializer=random_ops.random_uniform(dyn_shape,
|
| /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
82 dyn_shape.set_dynamic_dimension(0, true); in AcceptDynamicR1()
83 auto p0 = xla::Parameter(&builder, 0, dyn_shape, "P0"); in AcceptDynamicR1()
84 auto p1 = xla::Parameter(&builder, 1, dyn_shape, "P1"); in AcceptDynamicR1()
91 xla::Shape dyn_shape; in AcceptDynamicR2() local
92 dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {2, 4}); in AcceptDynamicR2()
93 dyn_shape.set_dynamic_dimension(1, true); in AcceptDynamicR2()
94 auto p0 = xla::Parameter(&builder, 0, dyn_shape, "P0"); in AcceptDynamicR2()
118 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {4}); in AcceptDynamicR1Tuple() local
119 dyn_shape.set_dynamic_dimension(0, true); in AcceptDynamicR1Tuple()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 1105 SmallVector<Value, 2> dyn_shape; in GetDotOpInitTensorDynSizes() local
1109 dyn_shape.push_back(b.create<DimOp>(loc, lhs, 0)); in GetDotOpInitTensorDynSizes()
1111 dyn_shape.push_back(b.create<DimOp>(loc, rhs, 1)); in GetDotOpInitTensorDynSizes()
1116 dyn_shape.push_back(b.create<DimOp>(loc, lhs, 0)); in GetDotOpInitTensorDynSizes()
1125 return dyn_shape; in GetDotOpInitTensorDynSizes()
1144 SmallVector<Value, 2> dyn_shape = GetDotOpInitTensorDynSizes( in matchAndRewrite() local
1146 auto init_tensor = GetInitTensor(rewriter, loc, shaped_type, dyn_shape); in matchAndRewrite()
1181 SmallVector<Value, 8> dyn_shape; in GetDotGeneralOpInitTensorDynSizes() local
1183 dyn_shape.push_back(b.create<DimOp>(loc, lhs, 0)); in GetDotGeneralOpInitTensorDynSizes()
1185 dyn_shape.push_back(b.create<DimOp>(loc, lhs, 1)); in GetDotGeneralOpInitTensorDynSizes()
[all …]
|