Searched refs:concatenated_tensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_utils_v1_test.py | 428 concatenated_tensor = training_utils_v1._append_composite_tensor(
430 evaluated_tensor = self.evaluate(concatenated_tensor)
438 concatenated_tensor = training_utils_v1._append_composite_tensor(
440 self.assertAllEqual(concatenated_tensor.indices, [[0, 0], [1, 0]])
441 self.assertAllEqual(concatenated_tensor.values, [1, 2])
442 self.assertAllEqual(concatenated_tensor.dense_shape, [2, 1])
449 concatenated_tensor = training_utils_v1._append_composite_tensor(
451 evaluated_tensor = self.evaluate(concatenated_tensor)
461 concatenated_tensor = training_utils_v1._append_composite_tensor(
464 self.assertAllEqual(concatenated_tensor.values, [0, 1, 2, 3, 4, 5])
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 786 Tensor concatenated_tensor; in OutputBatch() local
790 TF_RETURN_IF_ERROR(Concat<type>(context, tensors, &concatenated_tensor)); \ in OutputBatch()
791 context->set_output(0, concatenated_tensor); \ in OutputBatch()
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 370 Tensor concatenated_tensor; in ConcatInputTensors() local
372 Concat(context, to_concatenate, &concatenated_tensor); in ConcatInputTensors()
374 concatenated_tensors->push_back(concatenated_tensor); in ConcatInputTensors()
|