Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/framework/
Dcommon_shapes.py514 def _broadcast_shape_helper(shape_x, shape_y): argument
529 reversed(shape_y.dims),
562 def is_broadcast_compatible(shape_x, shape_y): argument
573 if shape_x.ndims is None or shape_y.ndims is None:
575 return _broadcast_shape_helper(shape_x, shape_y) is not None
578 def broadcast_shape(shape_x, shape_y): argument
591 if shape_x.ndims is None or shape_y.ndims is None:
593 return_dims = _broadcast_shape_helper(shape_x, shape_y)
596 % (shape_x, shape_y))
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor_shape.py408 def broadcast_dynamic_shape(shape_x, shape_y): argument
422 if not isinstance(shape_y, RaggedTensorDynamicShape):
426 if shape_x.rank is None or shape_y.rank is None:
428 broadcast_rank = max(shape_x.rank, shape_y.rank)
430 shape_y = shape_y.broadcast_to_rank(broadcast_rank)
434 shape_x = shape_x.broadcast_dimension(axis, shape_y.dimension_size(axis))
435 shape_y = shape_y.broadcast_dimension(axis, shape_x.dimension_size(axis))
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.h289 ShapeHandle shape_y,
Dcommon_shape_fns.cc1314 ShapeHandle shape_y, in BroadcastBinaryOpOutputShapeFnHelper() argument
1317 if (!c->RankKnown(shape_x) || !c->RankKnown(shape_y)) { in BroadcastBinaryOpOutputShapeFnHelper()
1322 const int32 rank_y = c->Rank(shape_y); in BroadcastBinaryOpOutputShapeFnHelper()
1337 dim_y_is_one ? dim_one : c->Dim(shape_y, i - (rank_out - rank_y)); in BroadcastBinaryOpOutputShapeFnHelper()
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py233 def broadcast_dynamic_shape(shape_x, shape_y): argument
254 return gen_array_ops.broadcast_args(shape_x, shape_y)
258 def broadcast_static_shape(shape_x, shape_y): argument
281 return common_shapes.broadcast_shape(shape_x, shape_y)
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc1740 ShapeHandle shape_y = c->input(1); in __anon7c94107b2902() local
1742 TF_RETURN_IF_ERROR(c->WithRank(shape_y, 1, &unused)); in __anon7c94107b2902()
1745 !c->ValueKnown(c->Dim(shape_y, 0))) { in __anon7c94107b2902()
1751 int64 y_dim = c->Value(c->Dim(shape_y, 0)); in __anon7c94107b2902()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.pbtxt473 argspec: "args=[\'shape_x\', \'shape_y\'], varargs=None, keywords=None, defaults=None"
477 argspec: "args=[\'shape_x\', \'shape_y\'], varargs=None, keywords=None, defaults=None"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt909 argspec: "args=[\'shape_x\', \'shape_y\'], varargs=None, keywords=None, defaults=None"
913 argspec: "args=[\'shape_x\', \'shape_y\'], varargs=None, keywords=None, defaults=None"