Searched defs:broadcast_to (Results 1 – 8 of 8) sorted by relevance
| /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/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()
|