Searched refs:unstacked (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | unstack_op_test.py | 149 unstacked = self.evaluate(array_ops.unstack(a))
151 self.assertEqual(len(unstacked), 2)
152 self.assertAllEqual(unstacked[0], [1, 2, 3])
153 self.assertAllEqual(unstacked[1], [4, 5, 6])
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils.cc | 183 Status StackTensor(WrappedTensor* unstacked, TensorDesc* result);
416 Status Vectorization::StackTensor(WrappedTensor* unstacked, in StackTensor() argument
418 if (unstacked->node->output_type(unstacked->output_index) == DT_VARIANT) { in StackTensor()
424 if (unstacked->stacked) { in StackTensor()
473 .Input(unstacked->node) // input in StackTensor()
491 .Input(unstacked->node) // input in StackTensor()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | decode_csv_vectorizer.cc | 36 TF_RETURN_IF_ERROR(inputs.unstacked(i, &defaults[i - 1])); in Vectorize()
|
| D | transpose_vectorizer.cc | 52 TF_RETURN_IF_ERROR(inputs.unstacked(1, &original_perm)); in Vectorize()
|
| D | reshape_vectorizer.cc | 58 TF_RETURN_IF_ERROR(inputs.unstacked(1, &shape)); in Vectorize()
|
| D | vectorizer.h | 60 Status unstacked(int index, T* result) const { in unstacked() function
|
| D | parse_single_example_vectorizer.cc | 38 TF_RETURN_IF_ERROR(inputs.unstacked(i, &dense_defaults[i - 1])); in Vectorize()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 253 unstacked = array_ops.unstack(tensor, axis=1)
254 for i in range(len(unstacked)):
256 dense_floats.append(array_ops.reshape(unstacked[i], [-1, 1]))
|