Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor_shape.py408 def broadcast_dynamic_shape(shape_x, shape_y): argument
420 if not isinstance(shape_x, RaggedTensorDynamicShape):
426 if shape_x.rank is None or shape_y.rank is None:
428 broadcast_rank = max(shape_x.rank, shape_y.rank)
429 shape_x = shape_x.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))
437 return shape_x
/external/tensorflow/tensorflow/python/framework/
Dcommon_shapes.py514 def _broadcast_shape_helper(shape_x, shape_y): argument
528 reversed(shape_x.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/core/framework/
Dcommon_shape_fns.h288 ShapeHandle shape_x,
Dcommon_shape_fns.cc1313 ShapeHandle shape_x, in BroadcastBinaryOpOutputShapeFnHelper() argument
1317 if (!c->RankKnown(shape_x) || !c->RankKnown(shape_y)) { in BroadcastBinaryOpOutputShapeFnHelper()
1321 const int32 rank_x = c->Rank(shape_x); in BroadcastBinaryOpOutputShapeFnHelper()
1334 : c->Dim(shape_x, i - (rank_out - rank_x)); 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.cc1739 ShapeHandle shape_x = c->input(0); in __anon7c94107b2902() local
1741 TF_RETURN_IF_ERROR(c->WithRank(shape_x, 1, &unused)); in __anon7c94107b2902()
1744 if (!c->ValueKnown(c->Dim(shape_x, 0)) || in __anon7c94107b2902()
1750 int64 x_dim = c->Value(c->Dim(shape_x, 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"