Searched defs:shape_x (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| 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
|
| /external/pytorch/test/inductor/ |
| D | indirect_assert_helper.py | 59 shape_x = [3, 2, 4] if dims == "3" else [3, 2] variable
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fold_broadcast.cc | 88 SmallVectorImpl<int64_t>& result_shape) { in RewriteBatchMatMulV2Op()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_shape.py | 443 def broadcast_dynamic_shape(shape_x, shape_y): argument
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | FullyConnectedLayerFixture.h | 141 const size_t shape_x = reshaped_weights_shape.x(); in compute_target() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 512 def broadcast_dynamic_shape(shape_x, shape_y): argument
546 def broadcast_static_shape(shape_x, shape_y): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1927 ShapeHandle shape_x = c->input(0); in __anon847f0b682902() local
|
| /external/pytorch/test/ |
| 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
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2083 ShapeHandle shape_x, in BroadcastBinaryOpOutputShapeFnHelper()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 311 auto shape_x = x.getShape(); in GetStaticBroadcastType() local
735 auto shape_x = x_rankless.getShape(); in AreBroadcastCompatible() local
|