Searched defs:broadcast_to (Results 1 – 13 of 13) sorted by relevance
| /external/executorch/kernels/portable/cpu/util/ |
| D | broadcast_util.cpp | 105 const Tensor& broadcast_to) { in tensor_is_broadcastable_to()
142 const Tensor& broadcast_to) { in broadcast_tensor()
|
| /external/executorch/kernels/portable/cpu/util/test/ |
| D | broadcast_test.cpp | 170 Tensor broadcast_to = tf.zeros({2, 2, 3, 4}); in TEST() local
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_shape.py | 475 def broadcast_to(rt_input, shape, broadcast_inner_dimensions=True): function
|
| D | ragged_array_ops.py | 1077 def broadcast_to( function
|
| D | dynamic_ragged_shape.py | 1623 def broadcast_to(rt_input, shape: DynamicRaggedShape): function
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 1034 def broadcast_to(array, shape): # pylint: disable=redefined-outer-name function
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/ |
| D | quantize_model_test.cc | 1367 const auto& broadcast_to = subgraph->operators[0]; in TEST_P() local
|
| /external/pytorch/torch/_numpy/ |
| D | _funcs_impl.py | 710 def broadcast_to(array: ArrayLike, shape, subok: NotImplementedType = False): function
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_model_test.cc | 1874 const auto& broadcast_to = subgraph->operators[0]; in TEST_P() local
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 2771 def broadcast_to(self, tensor, destinations): member in StrategyExtendedV1
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 407 static Value Broadcast1DToFeatureDim(Location loc, Value broadcast_to, in Broadcast1DToFeatureDim()
427 static Value BroadcastToShapeOf(Location loc, Value input, Value broadcast_to, in BroadcastToShapeOf()
|
| /external/pytorch/torch/_refs/ |
| D | __init__.py | 2705 def broadcast_to(a: TensorLikeType, size: ShapeType) -> TensorLikeType: function
|
| /external/pytorch/torch/onnx/ |
| D | symbolic_opset9.py | 864 def broadcast_to(g: jit_utils.GraphContext, self, size): function
|