Searched refs:tensor_is_broadcastable_to (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/test/ |
| H A D | broadcast_test.cpp | 29 using torch::executor::tensor_is_broadcastable_to;
69 ASSERT_TRUE(tensor_is_broadcastable_to(a, c)); in TEST()
74 ASSERT_TRUE(tensor_is_broadcastable_to(b, c)); in TEST()
93 ASSERT_FALSE(tensor_is_broadcastable_to(a, b)); in TEST()
99 ASSERT_FALSE(tensor_is_broadcastable_to(b, c)); in TEST()
109 EXPECT_FALSE(tensor_is_broadcastable_to(a, b)); in TEST()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | broadcast_util.cpp | 78 bool tensor_is_broadcastable_to( in tensor_is_broadcastable_to() function
103 bool tensor_is_broadcastable_to( in tensor_is_broadcastable_to() function
106 return tensor_is_broadcastable_to( in tensor_is_broadcastable_to()
166 tensor_is_broadcastable_to(broadcast_from, broadcast_to); in broadcast_tensor()
|
| H A D | broadcast_util.h | 26 bool tensor_is_broadcastable_to(
40 bool tensor_is_broadcastable_to(
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_copy.cpp | 38 ctx, tensor_is_broadcastable_to(src, in), InvalidArgument, out); in copy_out()
74 ctx, tensor_is_broadcastable_to(src, in), InvalidArgument, in); in copy_()
|
| H A D | op_index_put.cpp | 54 ctx, tensor_is_broadcastable_to(values, out), InvalidArgument, out); in index_put_out()
84 tensor_is_broadcastable_to(values.sizes(), {x_sizes, x_dim}), in index_put_out()
|
| H A D | op_addmm.cpp | 46 ctx, tensor_is_broadcastable_to(in, out), InvalidArgument, out); in addmm_out()
|