Searched refs:concatenated_tensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_utils_v1_test.py | 396 concatenated_tensor = training_utils_v1._append_composite_tensor(
398 evaluated_tensor = self.evaluate(concatenated_tensor)
406 concatenated_tensor = training_utils_v1._append_composite_tensor(
408 self.assertAllEqual(concatenated_tensor.indices, [[0, 0], [1, 0]])
409 self.assertAllEqual(concatenated_tensor.values, [1, 2])
410 self.assertAllEqual(concatenated_tensor.dense_shape, [2, 1])
417 concatenated_tensor = training_utils_v1._append_composite_tensor(
419 evaluated_tensor = self.evaluate(concatenated_tensor)
429 concatenated_tensor = training_utils_v1._append_composite_tensor(
432 self.assertAllEqual(concatenated_tensor.values, [0, 1, 2, 3, 4, 5])
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 441 Tensor concatenated_tensor; in ConcatInputTensors() local
443 Concat(context, to_concatenate, &concatenated_tensor); in ConcatInputTensors()
445 concatenated_tensors->push_back(concatenated_tensor); in ConcatInputTensors()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 859 Tensor concatenated_tensor; in OutputBatch() local
863 TF_RETURN_IF_ERROR(Concat<type>(context, tensors, &concatenated_tensor)); \ in OutputBatch()
864 context->set_output(0, concatenated_tensor); \ in OutputBatch()
|