Home
last modified time | relevance | path

Searched refs:broadcast_to (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dbroadcast_to_ops_test.py37 v_tf = array_ops.broadcast_to(constant_op.constant(x), [3, 3])
38 v_np = np.broadcast_to(x, [3, 3])
45 v_tf = array_ops.broadcast_to(constant_op.constant(x), [3, 3])
46 v_np = np.broadcast_to(x, [3, 3])
53 v_tf = array_ops.broadcast_to(constant_op.constant(x), [3, 3])
54 v_np = np.broadcast_to(x, [3, 3])
65 v_tf = array_ops.broadcast_to(constant_op.constant(x), output_shape)
66 v_np = np.broadcast_to(x, output_shape)
75 v_tf = array_ops.broadcast_to(constant_op.constant(x), output_shape)
76 v_np = np.broadcast_to(x, output_shape)
[all …]
Dlu_op_test.py112 array_ops.broadcast_to(
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_batch_gather_with_default_op.py118 pad = ragged_tensor_shape.broadcast_to(
128 pad = array_ops.broadcast_to(default_value, pad_shape)
Dragged_tensor_shape.py440 def broadcast_to(rt_input, shape, broadcast_inner_dimensions=True): function
475 return array_ops.broadcast_to(rt_input, shape.inner_dim_sizes)
Dragged_util.py172 repeats = array_ops.broadcast_to(repeats, broadcast_shape)
Dragged_dispatch.py206 x = ragged_tensor_shape.broadcast_to(
208 y = ragged_tensor_shape.broadcast_to(
Dragged_tensor_shape_test.py422 result = ragged_tensor_shape.broadcast_to(x, shape)
Dragged_tensor.py1265 default_value = array_ops.broadcast_to(default_value, value_shape)
/external/tensorflow/tensorflow/contrib/opt/python/training/
Dexternal_optimizer.py101 left_packed_bounds.extend(list(np.broadcast_to(bounds[0], shape).flat))
102 right_packed_bounds.extend(list(np.broadcast_to(bounds[1], shape).flat))
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_BroadcastTo.pbtxt32 >>> y = tf.broadcast_to(x, [3, 3])
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-strategy-extended.pbtxt42 name: "broadcast_to"
Dtensorflow.pbtxt480 name: "broadcast_to"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-strategy-extended.pbtxt42 name: "broadcast_to"
Dtensorflow.pbtxt916 name: "broadcast_to"
/external/tensorflow/tensorflow/compiler/tests/
Dbinary_ops_test.py1472 array_ops.broadcast_to,
1477 array_ops.broadcast_to,
1482 array_ops.broadcast_to,
1487 array_ops.broadcast_to,
1492 array_ops.broadcast_to,
1497 array_ops.broadcast_to,
1504 array_ops.broadcast_to,
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py218 return array_ops.broadcast_to(x, shape, name=name)
/external/tensorflow/tensorflow/python/distribute/
Ddistribute_lib.py1014 def broadcast_to(self, tensor, destinations): member in DistributionStrategyExtended
Dvalues.py625 return strategy.extended.broadcast_to(
/external/tensorflow/tensorflow/python/keras/layers/
Dwrappers_test.py174 np.broadcast_to(np.array([0, 1]), (1, 10, 2)))
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py1487 dilation_rate=np.broadcast_to(rate, (2,)),