Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/python/kernel_tests/
Dunstack_op_test.py53 cs = array_ops.unstack(x, num=shape[0])
74 cs = array_ops.unstack(x, num=shape[0])
88 cs = array_ops.unstack(x, num=shape[0])
102 cs = array_ops.unstack(x, num=shape[1], axis=1)
111 cs = array_ops.unstack(x)
120 array_ops.unstack(x)
125 array_ops.unstack(x, num=2)
132 array_ops.unstack(x)
143 actual_unstack = self.evaluate(array_ops.unstack(a, axis=j))
149 unstacked = self.evaluate(array_ops.unstack(a))
[all …]
Dtensor_array_ops_test.py225 w0 = ta.unstack(convert([1.0, 2.0, 3.0]))
236 w1 = ta.unstack(convert([[1.0, 1.1], [2.0, 2.1], [3.0, 3.1]]))
247 w2 = ta.unstack(convert([[], [], []]))
798 w_readonce = ta_readonce.unstack(value)
812 w_readtwice = ta_readtwice.unstack(value)
829 w = ta.unstack(value)
882 w = ta.unstack(value)
1213 w0 = ta.unstack(value)
1289 w = ta.unstack(x)
1318 w0 = ta.unstack(x)
[all …]
/external/tensorflow/tensorflow/contrib/gan/python/eval/python/
Dsummaries_impl.py120 image_list = (array_ops.unstack(gan_model.generator_inputs[:1]) +
121 array_ops.unstack(gan_model.generated_data[:1]) +
122 array_ops.unstack(reconstructions[:1]))
166 array_ops.unstack(gan_model.generator_inputs[:num_comparisons]))
168 array_ops.unstack(gan_model.generated_data[:num_comparisons]))
169 image_list.extend(array_ops.unstack(gan_model.real_data[:num_comparisons]))
171 generated_list = array_ops.unstack(
173 real_list = array_ops.unstack(gan_model.real_data[:num_comparisons])
Dclassifier_metrics_impl.py356 logits = array_ops.concat(array_ops.unstack(logits), 0)
519 real_a = array_ops.concat(array_ops.unstack(real_a), 0)
520 gen_a = array_ops.concat(array_ops.unstack(gen_a), 0)
910 real_a = array_ops.concat(array_ops.unstack(real_a), 0)
911 gen_a = array_ops.concat(array_ops.unstack(gen_a), 0)
Deval_utils_test.py36 images=array_ops.unstack(array_ops.zeros([25, 32, 32, 3])),
Deval_utils_impl.py113 images = array_ops.unstack(images)
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
Dlstm_ops.py228 return array_ops.unstack(i), array_ops.unstack(cs), array_ops.unstack(
229 f), array_ops.unstack(o), array_ops.unstack(ci), array_ops.unstack(
230 co), array_ops.unstack(h)
574 outputs = array_ops.unstack(outputs)
Dfused_rnn_cell.py115 outputs = array_ops.unstack(outputs)
118 inputs = array_ops.unstack(inputs)
/external/tensorflow/tensorflow/contrib/learn/python/learn/ops/
Dseq2seq_ops.py85 in_x = array_ops.unstack(x, axis=1)
86 y = array_ops.unstack(y, axis=1)
/external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
Dresnet_preprocessing.py70 offset_y, offset_x, _ = tf.unstack(bbox_begin)
71 target_height, target_width, _ = tf.unstack(bbox_size)
/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/
Dbatch_ops_utils.py72 return array_ops.unstack(values)
78 sizes = array_ops.unstack(sizes)
/external/tensorflow/tensorflow/compiler/tests/
Dtensor_array_ops_test.py138 w0 = ta.unstack(convert([1.0, 2.0, 3.0]))
152 w1 = ta.unstack(convert([[1.0, 1.1], [2.0, 2.1], [3.0, 3.1]]))
168 w2 = ta.unstack(convert([[], [], []]))
540 w_readtwice = ta_readtwice.unstack(value)
557 w = ta.unstack(value)
849 w0 = ta.unstack(value)
902 w = ta.unstack(x)
931 ta = ta.unstack(array_ops.zeros([0, 3, 5]))
1012 w = ta.unstack(values)
/external/tensorflow/tensorflow/python/ops/
Dtensor_array_ops.py334 def unstack(self, value, name=None): member in _GraphTensorArray
621 def unstack(self, value, name=None): member in _GraphTensorArrayV2
896 def unstack(self, value, name=None): member in _EagerTensorArray
898 tensors = array_ops.unstack(value, name=name)
911 for index, val in zip(indices, array_ops.unstack(value)):
1178 def unstack(self, value, name=None): member in TensorArray
1195 return self._implementation.unstack(value, name=name)
Dfunctional_ops.py102 infer_shape=True).unstack(elem)
211 infer_shape=True).unstack(elem)
413 elem_ta.unstack(elem) for elem_ta, elem in zip(elems_ta, elems_flat)]
/external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
Dfused_rnn_cell_test.py51 unpacked_inputs = array_ops.unstack(inputs)
122 unpacked_inputs = array_ops.unstack(inputs)
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch.py253 unstacked = array_ops.unstack(tensor, axis=1)
919 active_handlers = array_ops.unstack(active_handlers, axis=0)
1098 sizes = array_ops.unstack(split_sizes)
1228 gradients_list = array_ops.unstack(
1246 hessian_row = array_ops.squeeze(array_ops.unstack(hessian_row), [0])
1254 gradients_list = array_ops.unstack(
1271 hessian_row = array_ops.squeeze(array_ops.unstack(hessian_row), [0])
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Dunidirectional_sequence_rnn_test.py83 outputs = tf.unstack(outputs, axis=0)
85 rnn_input = tf.unstack(x, self.time_steps, 1)
Dunidirectional_sequence_lstm_test.py87 outputs = tf.unstack(outputs, axis=0)
89 lstm_input = tf.unstack(x, self.time_steps, 1)
Dbidirectional_sequence_lstm_test.py93 output = tf.unstack(output, axis=0)
96 lstm_input = tf.unstack(x, self.time_steps, 1)
Dbidirectional_sequence_rnn_test.py103 output = tf.unstack(output, axis=0)
106 rnn_inputs = tf.unstack(x, self.time_steps, 1)
/external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_colorbot/
Drnn_colorbot.py188 chars = tf.unstack(chars, axis=0)
303 preds = tf.unstack(model((chars, length), training=False)[0])
/external/tensorflow/tensorflow/python/ops/parallel_for/
Darray_test.py158 return array_ops.unstack(
159 x_i, 4, axis=-1), array_ops.unstack(
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-tensor-array.pbtxt62 name: "unstack"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-tensor-array.pbtxt62 name: "unstack"
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dstate_saving_rnn_estimator.py375 return array_ops.unstack(sequence_features, axis=1)
380 return array_ops.unstack(

123