Searched refs:extra_outputs (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | trt_mode_test.py | 106 extra_outputs=[])
150 extra_outputs=[[[2, 3]], [[4, 6]]],
|
| D | batch_matmul_test.py | 47 extra_outputs=[])
|
| D | tf_trt_integration_test_base.py | 211 input_mask, output_mask, extra_inputs, extra_outputs): argument
247 for extra_in_shape, extra_out_shape in zip(extra_inputs, extra_outputs):
262 expected_output_dims=[output_shapes] + extra_outputs)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | cond_v2.py | 1075 for branch_graph, extra_outputs in zip(branch_graphs, extra_branch_outputs):
1076 branch_graph.outputs.extend(extra_outputs)
|