Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dunstack_op_test.py80 cs = array_ops.unstack(x, axis=axis)
105 cs = array_ops.unstack(x, axis=axis)
120 return array_ops.unstack(x, num=shape[0])[i]
134 return array_ops.unstack(x, num=shape[1], axis=1)[i]
144 cs = array_ops.unstack(x)
154 array_ops.unstack(x)
160 array_ops.unstack(x, num=2)
168 array_ops.unstack(x)
179 actual_unstack = self.evaluate(array_ops.unstack(a, axis=j))
185 unstacked = self.evaluate(array_ops.unstack(a))
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Dtensor_array_ops_test.py150 w0 = ta.unstack(convert([1.0, 2.0, 3.0]))
167 w1 = ta.unstack(
187 w2 = ta.unstack(convert([[], [], []]))
662 w_readtwice = ta_readtwice.unstack(value)
683 w = ta.unstack(value)
968 w = ta.unstack(x)
1003 ta = ta.unstack(array_ops.zeros([0, 3, 5]))
1016 ta = ta.unstack(array_ops.zeros([0, 3, 5]))
1100 w = ta.unstack(values)
Dunstack_test.py35 ret = array_ops.unstack(x_tf)
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dunpack.py49 outs = tf.unstack(input_tensor, axis=get_valid_axis(parameters))
Dunidirectional_sequence_rnn.py53 outs = tf.unstack(outs, axis=1)
Dunidirectional_sequence_lstm.py55 outs = tf.unstack(outs, axis=1)
/external/tensorflow/tensorflow/python/keras/
Dconstraints.py234 K.stack(array_ops.unstack(w, axis=-1), axis=0))
235 return K.reshape(K.stack(array_ops.unstack(w, axis=0), axis=-1),
/external/tensorflow/tensorflow/python/ops/
Dtensor_array_ops.py323 def unstack(self, value, name=None): member in _GraphTensorArray
586 def unstack(self, value, name=None): member in _GraphTensorArrayV2
881 def unstack(self, value, name=None): member in _EagerTensorArray
883 tensors = array_ops.unstack(value, name=name)
896 for index, val in zip(indices, array_ops.unstack(value)):
1204 def unstack(self, value, name=None): member in TensorArray
1221 return self._implementation.unstack(value, name=name)
Dfunctional_ops.py111 infer_shape=True).unstack(elem)
306 infer_shape=True).unstack(elem)
602 elem_ta.unstack(elem) for elem_ta, elem in zip(elems_ta, elems_flat)
Dmap_fn.py455 ta.unstack(t) for (ta, t) in zip(elems_batchable_ta, elems_batchable)
/external/tensorflow/tensorflow/python/kernel_tests/
Dtensor_array_ops_test.py269 w0 = ta.unstack(convert([1.0, 2.0, 3.0]))
280 w1 = ta.unstack(convert([[1.0, 1.1], [2.0, 2.1], [3.0, 3.1]]))
291 w2 = ta.unstack(convert([[], [], []]))
848 w_readonce = ta_readonce.unstack(value)
862 w_readtwice = ta_readtwice.unstack(value)
879 w = ta.unstack(value)
932 w = ta.unstack(value)
1274 w0 = ta.unstack(value)
1349 w = ta.unstack(x)
1378 w0 = ta.unstack(x)
[all …]
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Dbidirectional_sequence_rnn_test.py122 output = tf.unstack(output, axis=0)
125 rnn_inputs = tf.unstack(x, self.time_steps, 1)
Dunidirectional_sequence_lstm_test.py102 outputs = tf.unstack(outputs, axis=0)
104 lstm_input = tf.unstack(x, self.time_steps, 1)
Dunidirectional_sequence_rnn_test.py98 outputs = tf.unstack(outputs, axis=0)
100 rnn_input = tf.unstack(x, self.time_steps, 1)
Dbidirectional_sequence_lstm_test.py110 output = tf.unstack(output, axis=0)
113 lstm_input = tf.unstack(x, self.time_steps, 1)
/external/tensorflow/tensorflow/python/keras/engine/
Dkeras_tensor_test.py86 kts = array_ops.unstack(kt)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-tensor-array.pbtxt70 name: "unstack"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-tensor-array.pbtxt70 name: "unstack"
/external/tensorflow/tensorflow/lite/python/
Dconvert_test.py344 a0, a1 = array_ops.unstack(a)
345 b0, b1 = array_ops.unstack(b)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dflat_map_test.py139 .unstack(math_ops.range(i)))
/external/tensorflow/tensorflow/python/ops/parallel_for/
Darray_test.py238 return array_ops.unstack(
239 x_i, 4, axis=-1), array_ops.unstack(
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_tridiag.py291 superdiag, diag, subdiag = array_ops.unstack(
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_tensor_spec_test.py259 *[array_ops.unstack(tensor) for tensor in batched_tensor_list])
/external/tensorflow/tensorflow/python/framework/
Dsparse_tensor.py389 pieces = array_ops.unstack(dense_shape, num=self._shape.rank)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_vectorization_test.py326 map_fns = [array_ops.unstack, lambda x: array_ops.unstack(x, axis=-1)]

123