Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/python/kernel_tests/
Dbroadcast_to_ops_test.py38 v_tf = array_ops.broadcast_to(constant_op.constant(x), [3, 3])
39 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])
52 v_tf = array_ops.broadcast_to(constant_op.constant(x), [3, 3])
53 v_np = np.broadcast_to(x, [3, 3])
63 v_tf = array_ops.broadcast_to(constant_op.constant(x), output_shape)
64 v_np = np.broadcast_to(x, output_shape)
72 v_tf = array_ops.broadcast_to(constant_op.constant(x), output_shape)
73 v_np = np.broadcast_to(x, output_shape)
[all …]
Dbatch_matmul_op_test.py259 array_ops.broadcast_to(matrix_a, broadcasted_a_shape),
260 array_ops.broadcast_to(matrix_b, broadcasted_b_shape)),
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_BroadcastTo.pbtxt32 >>> y = tf.broadcast_to(x, [3, 3])
46 However, `broadcast_to` does not carry with it any such benefits.
48 shape. (In a graph context, `broadcast_to` might be fused to
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_tridiag.py320 rhs = array_ops.broadcast_to(
324 diagonals = array_ops.broadcast_to(
328 diagonals = array_ops.broadcast_to(
333 array_ops.broadcast_to(d, array_ops.concat(
348 return array_ops.broadcast_to(
Dlinear_operator_permutation.py223 x = array_ops.broadcast_to(x, broadcast_x_shape)
224 perm = array_ops.broadcast_to(perm, broadcast_shape)
Dlinear_operator_toeplitz.py254 row = array_ops.broadcast_to(row, total_shape)
255 col = array_ops.broadcast_to(col, total_shape)
Dlinear_operator_util.py348 batch_matrices[i] = array_ops.broadcast_to(mat, bcast_shape)
358 batch_matrices[i] = array_ops.broadcast_to(
Dlinalg_impl.py987 broadcast_rhs = array_ops.broadcast_to(rhs, rhs_broadcast_shape)
991 broadcast_perm = array_ops.broadcast_to(perm, rhs_broadcast_shape[:-1])
994 broadcast_batch_indices = array_ops.broadcast_to(
1140 batch_indices = array_ops.broadcast_to(
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_where_op.py211 condition = ragged_tensor_shape.broadcast_to(condition, shape)
212 x = ragged_tensor_shape.broadcast_to(x, shape)
213 y = ragged_tensor_shape.broadcast_to(y, shape)
Dragged_batch_gather_with_default_op.py115 pad = ragged_tensor_shape.broadcast_to(
125 pad = array_ops.broadcast_to(default_value, pad_shape)
Dragged_tensor_shape.py476 def broadcast_to(rt_input, shape, broadcast_inner_dimensions=True): function
511 return array_ops.broadcast_to(rt_input, shape.inner_dim_sizes)
590 array_ops.broadcast_to(rt_input.flat_values, new_shape))
Dragged_dispatch.py220 x = ragged_tensor_shape.broadcast_to(
222 y = ragged_tensor_shape.broadcast_to(
Dragged_tensor_shape_test.py418 result = ragged_tensor_shape.broadcast_to(x, shape)
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlegalize-tf.mlir1691 // CHECK: [[BCT:%.*]] = "tfl.broadcast_to"(%arg0, %arg1) : (tensor<3xf32>, tensor<2xi32>) -> tenso…
1700 // CHECK: [[BCT:%.*]] = "tfl.broadcast_to"(%arg0, %arg1) : (tensor<3xi32>, tensor<2xi32>) -> tenso…
1735 // CHECK: [[BCT:%.*]] = "tfl.broadcast_to"(%arg0, [[CST]])
1736 // CHECK: [[BCT_0:%.*]] = "tfl.broadcast_to"(%arg1, [[CST]])
1737 // CHECK: [[BCT_1:%.*]] = "tfl.broadcast_to"(%arg2, [[CST]])
1749 // CHECK: [[BCT:%.*]] = "tfl.broadcast_to"(%arg1, [[CST]])
1760 // CHECK: [[BCT:%.*]] = "tfl.broadcast_to"(%arg0, [[CST]])
1761 // CHECK: [[BCT_0:%.*]] = "tfl.broadcast_to"(%arg1, [[CST]])
1768 // CHECK: [[BCAST:%.*]] = "tfl.broadcast_to"(%arg0, [[CST]])
1769 // CHECK: [[BCAST_1:%.*]] = "tfl.broadcast_to"(%arg1, [[CST]])
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dlinear_operator_permutation_test.py60 perm = array_ops.broadcast_to(perm, shape[:-1])
/external/tensorflow/tensorflow/python/ops/linalg/sparse/
Dconjugate_gradient.py121 r0 = array_ops.broadcast_to(rhs, broadcast_rhs_shape)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-strategy-extended.pbtxt43 name: "broadcast_to"
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Dsnapshot_dataset_benchmark.py50 lambda x: gen_array_ops.broadcast_to(x, [50, 50, 3]))
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_array_ops.py138 return array_ops.broadcast_to(fill_value, shape)
156 return array_ops.broadcast_to(fill_value, array_ops.shape(a))
1032 def broadcast_to(array, shape): # pylint: disable=redefined-outer-name function
1219 array_ops.broadcast_to(mask, array_ops.shape(m)), m, z)
1240 array_ops.broadcast_to(mask, array_ops.shape(m)), z, m)
1423 arr = array_ops.broadcast_to(arr, arr_shape)
1424 indices = array_ops.broadcast_to(indices, indices_shape)
1765 b = array_ops.broadcast_to(b, array_ops.shape(a))
Dnp_math_ops.py461 y = array_ops.broadcast_to(y, x.shape)
512 x1 = array_ops.broadcast_to(x1, shape)
513 x2 = array_ops.broadcast_to(x2, shape)
1287 weights_sum = np_array_ops.broadcast_to(weights_sum, array_ops.shape(avg))
/external/tensorflow/tensorflow/compiler/tests/
Dbinary_ops_test.py1349 np.broadcast_to(np.arange(0, 7, dtype=dtype), (3, 2, 1, 7)),
1356 expected=np.broadcast_to(
1365 np.broadcast_to(
1373 expected=np.broadcast_to(
1626 array_ops.broadcast_to,
1631 array_ops.broadcast_to,
1638 array_ops.broadcast_to,
1645 array_ops.broadcast_to,
/external/tensorflow/tensorflow/lite/kernels/
DAndroid.bp70 "broadcast_to.cc",
/external/tensorflow/tensorflow/python/ops/parallel_for/
Darray_test.py146 return (array_ops.broadcast_to(x1, [2, 2, 3]),
147 array_ops.broadcast_to(x1, [1, 2, 1, 3]))
/external/tensorflow/tensorflow/lite/
Dtflite_static.bp49 "kernels/broadcast_to.cc",
/external/tensorflow/tensorflow/python/distribute/
Dvalues_util.py327 return strategy.extended.broadcast_to(

123