| /external/tensorflow/tensorflow/python/framework/ |
| D | common_shapes.py | 25 def _broadcast_shape_helper(shape_x, shape_y): argument
40 reversed(shape_y.dims),
73 def is_broadcast_compatible(shape_x, shape_y): argument
84 if shape_x.ndims is None or shape_y.ndims is None:
86 return _broadcast_shape_helper(shape_x, shape_y) is not None
89 def broadcast_shape(shape_x, shape_y): argument
102 if shape_x.ndims is None or shape_y.ndims is None:
104 return_dims = _broadcast_shape_helper(shape_x, shape_y)
107 % (shape_x, shape_y))
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fold_broadcast.cc | 85 [&](ArrayRef<int64_t> shape_x, ArrayRef<int64_t> shape_y, in RewriteBatchMatMulV2Op()
87 if (shape_x.size() < 2 || shape_y.size() < 2) { in RewriteBatchMatMulV2Op()
95 shape_x.drop_back(2), shape_y.drop_back(2), result_shape)) { in RewriteBatchMatMulV2Op()
105 matmul_op.adj_y() ? shape_y.back() : *(shape_y.rbegin() + 1); in RewriteBatchMatMulV2Op()
107 !matmul_op.adj_y() ? shape_y.back() : *(shape_y.rbegin() + 1); in RewriteBatchMatMulV2Op()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_shape.py | 444 def broadcast_dynamic_shape(shape_x, shape_y): argument
458 if not isinstance(shape_y, RaggedTensorDynamicShape):
462 if shape_x.rank is None or shape_y.rank is None:
464 broadcast_rank = max(shape_x.rank, shape_y.rank)
466 shape_y = shape_y.broadcast_to_rank(broadcast_rank)
470 shape_x = shape_x.broadcast_dimension(axis, shape_y.dimension_size(axis))
471 shape_y = shape_y.broadcast_dimension(axis, shape_x.dimension_size(axis))
|
| D | ragged_where_op.py | 208 shape_y = ragged_tensor_shape.RaggedTensorDynamicShape.from_tensor(y)
210 shape_c, ragged_tensor_shape.broadcast_dynamic_shape(shape_x, shape_y))
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | chlo_ops.cc | 77 auto shape_y = y_ranked.getShape(); in GetBroadcastType() local
79 if (shape_x.size() == shape_y.size()) { in GetBroadcastType()
83 auto y_val = shape_y[i]; in GetBroadcastType()
93 auto shape_large = shape_x.size() > shape_y.size() ? shape_x : shape_y; in GetBroadcastType()
94 auto shape_small = shape_x.size() <= shape_y.size() ? shape_x : shape_y; in GetBroadcastType()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.h | 222 ShapeHandle shape_y,
|
| D | common_shape_fns.cc | 1993 ShapeHandle shape_y, in BroadcastBinaryOpOutputShapeFnHelper() argument
1997 if (!c->RankKnown(shape_x) || !c->RankKnown(shape_y)) { in BroadcastBinaryOpOutputShapeFnHelper()
2002 const int32 rank_y = c->Rank(shape_y); in BroadcastBinaryOpOutputShapeFnHelper()
2017 dim_y_is_one ? dim_one : c->Dim(shape_y, i - (rank_out - rank_y)); in BroadcastBinaryOpOutputShapeFnHelper()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 383 auto shape_y = y.getShape(); in GetStaticBroadcastType() local
385 if (shape_x.size() == shape_y.size()) { in GetStaticBroadcastType()
389 auto y_val = shape_y[i]; in GetStaticBroadcastType()
395 auto shape_large = shape_x.size() > shape_y.size() ? shape_x : shape_y; in GetStaticBroadcastType()
396 auto shape_small = shape_x.size() <= shape_y.size() ? shape_x : shape_y; in GetStaticBroadcastType()
863 auto shape_y = y_rankless.getShape(); in AreBroadcastCompatible() local
865 int rank_diff = shape_x.size() - shape_y.size(); in AreBroadcastCompatible()
868 for (int i = 0, s = std::min(shape_x.size(), shape_y.size()); i < s; i++) { in AreBroadcastCompatible()
871 if ((shape_x[index_x] == -1 && shape_y[index_y] != 1) || in AreBroadcastCompatible()
872 (shape_y[index_y] == -1 && shape_x[index_x] != 1)) { in AreBroadcastCompatible()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 506 def broadcast_dynamic_shape(shape_x, shape_y): argument
535 return gen_array_ops.broadcast_args(shape_x, shape_y)
540 def broadcast_static_shape(shape_x, shape_y): argument
571 return common_shapes.broadcast_shape(shape_x, shape_y)
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1895 ShapeHandle shape_y = c->input(1); in __anon42d741192902() local
1897 TF_RETURN_IF_ERROR(c->WithRank(shape_y, 1, &unused)); in __anon42d741192902()
1900 !c->ValueKnown(c->Dim(shape_y, 0))) { in __anon42d741192902()
1906 int64 y_dim = c->Value(c->Dim(shape_y, 0)); in __anon42d741192902()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 525 argspec: "args=[\'shape_x\', \'shape_y\'], varargs=None, keywords=None, defaults=None"
529 argspec: "args=[\'shape_x\', \'shape_y\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 945 argspec: "args=[\'shape_x\', \'shape_y\'], varargs=None, keywords=None, defaults=None"
949 argspec: "args=[\'shape_x\', \'shape_y\'], varargs=None, keywords=None, defaults=None"
|