Searched refs:tensor_slicing (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/debug/cli/ |
| H A D | command_parser.py | 162 tensor_slicing = in_str[in_str.index("["):]
165 tensor_slicing = ""
167 return tensor_name, tensor_slicing
467 def evaluate_tensor_slice(tensor, tensor_slicing): argument
484 if not validate_slicing_string(tensor_slicing):
487 return tensor[_parse_slices(tensor_slicing)]
|
| H A D | cli_shared.py | 149 tensor_slicing=None, argument
180 if tensor_slicing:
182 value = command_parser.evaluate_tensor_slice(tensor, tensor_slicing)
183 sliced_name = tensor_name + tensor_slicing
|
| H A D | command_parser_test.py | 208 tensor_slicing) = command_parser.parse_tensor_name_with_slicing(
212 self.assertEqual("", tensor_slicing)
216 tensor_slicing) = command_parser.parse_tensor_name_with_slicing(
220 self.assertEqual("[:, 1]", tensor_slicing)
|
| H A D | analyzer_cli.py | 929 tensor_name, tensor_slicing = (
993 tensor_slicing=tensor_slicing,
1041 tensor_slicing=tensor_slicing,
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/debug/wrappers/ |
| H A D | local_cli_wrapper.py | 492 tensor_name, tensor_slicing = (
514 tensor_slicing=tensor_slicing,
|