Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/kernel_tests/
Dsplit_op_test.py70 num_or_size_splits=constant_op.constant(2),
227 tf_ans = array_ops.split(value=x, num_or_size_splits=num, axis=dim)
250 tf_ans = array_ops.split(value=x, num_or_size_splits=num, axis=dim)
298 value=inp, num_or_size_splits=num_split, axis=split_dim))
320 s = array_ops.split(value=inp_tensor, num_or_size_splits=4, axis=1)
338 array_ops.split(value=[[0, 1], [2, 3]], num_or_size_splits=4, axis=2)
342 array_ops.split(value=[[0, 1], [2, 3]], num_or_size_splits=4, axis=-3)
346 array_ops.split(value=[0, 1, 2, 3], num_or_size_splits=3, axis=0)
351 num_or_size_splits=4,
359 num_or_size_splits=4,
Dmetrics_test.py1858 value=prec, num_or_size_splits=2, axis=0)
1862 value=rec, num_or_size_splits=2, axis=0)
1889 value=prec, num_or_size_splits=2, axis=0)
1893 value=rec, num_or_size_splits=2, axis=0)
1918 value=prec, num_or_size_splits=2, axis=0)
1920 value=rec, num_or_size_splits=2, axis=0)
Dcwise_ops_test.py1217 real, imag = array_ops.split(value=inx, num_or_size_splits=2, axis=1)
1256 xr, xi, yr, yi = array_ops.split(value=inp, num_or_size_splits=4, axis=1)
/external/tensorflow/tensorflow/python/tpu/
Dtpu_feed.py71 num_or_size_splits = [ceil_ratio] * num_full_slots + [left_over]
72 if len(num_or_size_splits) < dim:
73 num_or_size_splits += [0] * (dim - len(num_or_size_splits))
78 x, num_or_size_splits=num_or_size_splits, axis=axis))
/external/tensorflow/tensorflow/python/ops/
Drnn_grad_test.py145 w_i, w_c, w_f, w_o = array_ops.split(w, num_or_size_splits=4, axis=1)
146 b_i, b_c, b_f, b_o = array_ops.split(b, num_or_size_splits=4)
Darray_ops.py2072 def split(value, num_or_size_splits, axis=0, num=None, name="split"): argument
2126 if isinstance(num_or_size_splits,
2129 axis=axis, num_split=num_or_size_splits, value=value, name=name)
2131 size_splits = ops.convert_to_tensor(num_or_size_splits)
2136 "to split. Argument provided: %s" % (num_or_size_splits,))
2143 raise ValueError("Cannot infer num from shape %s" % num_or_size_splits)
Dimage_ops_impl.py5053 value=boxes_a, num_or_size_splits=4, axis=2)
5055 value=boxes_b, num_or_size_splits=4, axis=2)
5479 value=boxes, num_or_size_splits=4, axis=2)
Dgradients_test.py86 split_wx = array_ops.split(value=wx, num_or_size_splits=2, axis=0)
/external/tensorflow/tensorflow/compiler/tests/
Dlstm.py73 value=xmw, num_or_size_splits=4, axis=1)
Dbinary_ops_test.py1419 lambda x, y: array_ops.split(value=y, num_or_size_splits=3, axis=x),
1432 lambda x, y: array_ops.split(value=y, num_or_size_splits=2, axis=x),
1443 return array_ops.split(value=y, num_or_size_splits=[2, 3], axis=x)
/external/tensorflow/tensorflow/python/keras/premade/
Dwide_deep_test.py100 l1, l2 = array_ops.split(l, num_or_size_splits=2, axis=1)
104 h1, h2 = array_ops.split(h, num_or_size_splits=2, axis=1)
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_test.py262 array_ops.split(value=[1, 2, 3], num_or_size_splits=-1)
267 array_ops.split(value=[1, 2, 3], num_or_size_splits=0)
Dbackprop_test.py1362 x1, _ = array_ops.split(x, num_or_size_splits=2, axis=1)
/external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/
Drnn_cell_impl.py606 r, u = array_ops.split(value=value, num_or_size_splits=2, axis=1)
785 c, h = array_ops.split(value=state, num_or_size_splits=2, axis=one)
793 value=gate_inputs, num_or_size_splits=4, axis=one)
1057 value=lstm_matrix, num_or_size_splits=4, axis=1)
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy.py1586 num_or_size_splits=self.num_replicas_in_sync,
1599 num_or_size_splits=self.num_replicas_in_sync,
/external/tensorflow/tensorflow/lite/python/
Dutil_test.py109 value=in_tensor, num_or_size_splits=[1, 1, 1, 1], axis=0)
/external/tensorflow/tensorflow/python/keras/layers/
Drecurrent.py2438 self.kernel, num_or_size_splits=4, axis=1)
2445 self.bias, num_or_size_splits=4, axis=0)
2472 z = array_ops.split(z, num_or_size_splits=4, axis=1)
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinalg_impl.py1178 array_ops.shape(lower_upper)[-2:], num_or_size_splits=2)
/external/tensorflow/tensorflow/python/grappler/
Dauto_mixed_precision_test.py668 x1, x2, x3 = array_ops.split(x, num_or_size_splits=3, axis=3)
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining.py2821 num_or_size_splits=all_shapes,
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.pbtxt1037 …argspec: "args=[\'value\', \'num_or_size_splits\', \'axis\', \'num\', \'name\'], varargs=None, key…
/external/tensorflow/tensorflow/python/framework/
Dfunction_test.py1522 value=math_ops.matmul(xm, weights), num_or_size_splits=4, axis=1)
/external/tensorflow/tensorflow/python/keras/
Dbackend.py3299 num_or_size_splits=x_shape[axis],
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt2261 …argspec: "args=[\'value\', \'num_or_size_splits\', \'axis\', \'num\', \'name\'], varargs=None, key…
/external/tensorflow/
DRELEASE.md4148 `tf.split(value, num_or_size_splits, axis)`.