Searched defs:broadcast_to (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_shape.py | 476 def broadcast_to(rt_input, shape, broadcast_inner_dimensions=True): function
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 1032 def broadcast_to(array, shape): # pylint: disable=redefined-outer-name function
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_model_test.cc | 1749 const auto& broadcast_to = subgraph->operators[0]; in TEST_P() local
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 2655 def broadcast_to(self, tensor, destinations): member in StrategyExtendedV1
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 505 static Value Broadcast1DToFeatureDim(Location loc, Value broadcast_to, in Broadcast1DToFeatureDim()
525 static Value BroadcastToShapeOf(Location loc, Value input, Value broadcast_to, in BroadcastToShapeOf()
|