Searched refs:input_elem (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/cc/tools/ |
| D | freeze_saved_model.cc | 60 for (auto& input_elem : signature_def.inputs()) { in GetSignatureDefsInputsAndOutputs() local
61 GetTensorNamesFromTensorInfo(input_elem.second, inputs); in GetSignatureDefsInputsAndOutputs()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unique_op.cc | 92 auto input_elem = xla::DynamicSlice(input, {j}, {1}); in BuildInnerLoopBody() local
93 auto input_elem_scalar = xla::Reshape(single_element_shape, input_elem); in BuildInnerLoopBody()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | padding_fifo_queue_test.py | 1591 for (input_elem, output_elem) in zip(input_tuple, output_tuple):
1592 self.assertAllEqual(input_elem, output_elem)
|
| D | fifo_queue_test.py | 614 for (input_elem, output_elem) in zip(input_tuple, output_tuple):
615 self.assertAllEqual(input_elem, output_elem)
|