Searched defs:shape_y (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | common_shapes.py | 21 def _broadcast_shape_helper(shape_x, shape_y): argument
71 def is_broadcast_compatible(shape_x, shape_y): argument
87 def broadcast_shape(shape_x, shape_y): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | fold_broadcast.cc | 88 SmallVectorImpl<int64_t>& result_shape) { in RewriteBatchMatMulV2Op()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_tensor_shape.py | 443 def broadcast_dynamic_shape(shape_x, shape_y): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | array_ops.py | 512 def broadcast_dynamic_shape(shape_x, shape_y): argument
546 def broadcast_static_shape(shape_x, shape_y): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 6418 def helper(shape_x, shape_y, shape_z): argument
6593 def helper1(shape_x, shape_y, shape_z, shape_w): argument
6611 def helper(shape_x, shape_y, shape_z): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | array_ops.cc | 1928 ShapeHandle shape_y = c->input(1); in __anon38bbb0e82902() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns.cc | 2084 ShapeHandle shape_y, in BroadcastBinaryOpOutputShapeFnHelper()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 312 auto shape_y = y.getShape(); in GetStaticBroadcastType() local
736 auto shape_y = y_rankless.getShape(); in AreBroadcastCompatible() local
|