Home
last modified time | relevance | path

Searched defs:broadcast_from (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/executorch/kernels/portable/cpu/util/
H A Dbroadcast_util.cpp104 const Tensor& broadcast_from, in tensor_is_broadcastable_to()
141 const Tensor& broadcast_from, in broadcast_tensor()
312 const Tensor& broadcast_from) { in linearize_access_indexes()
/aosp_15_r20/external/executorch/kernels/portable/cpu/util/test/
H A Dbroadcast_test.cpp169 Tensor broadcast_from = tf.zeros({2, 1, 3, 1}); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc408 Value broadcast_from, int64_t feature_dim, in Broadcast1DToFeatureDim()