| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| graph_matcher_test.py | 78 self.assertEqual(match_result.get_tensor(inputs_pattern), inputs)
79 self.assertEqual(match_result.get_tensor('inputs'), inputs)
106 self.assertEqual(match_result.get_tensor(y1_pattern), y1)
151 self.assertEqual(match_result.get_tensor(reshape_pattern), reshape)
152 self.assertEqual(match_result.get_tensor('slice'), None)
162 self.assertEqual(match_result.get_tensor(reshape_pattern), reshape)
163 self.assertEqual(match_result.get_tensor('slice'), slicing)
|
| fold_batch_norms.py | 195 layer_tensor = match_result.get_tensor(layer_pattern)
210 input_tensor = match_result.get_tensor(input_pattern)
211 weight_tensor = match_result.get_tensor(weight_pattern)
212 gamma_tensor = match_result.get_tensor(gamma_pattern)
213 beta_tensor = match_result.get_tensor(beta_pattern)
250 bn_decay_mean_tensor = mul_match_result.get_tensor(bn_decay_pattern)
254 bn_decay_var_tensor = mul_match_result.get_tensor(bn_decay_pattern)
256 mean_tensor = match_result.get_tensor(mean_pattern)
257 variance_tensor = match_result.get_tensor(variance_pattern)
[all...] |
| quantize.py | 197 weight_tensor = match_result.get_tensor(weight_pattern)
199 weight_tensor = match_result.get_tensor(folded_weight_pattern)
|
| graph_matcher.py | 166 def get_tensor(self, pattern_or_name): member in class:MatchResult
|
| /external/tensorflow/tensorflow/contrib/summary/ |
| summary_ops_test.py | 187 self.assertEqual(2, get_tensor(self.db, x_id, 1))
188 self.assertEqual(3, get_tensor(self.db, y_id, 1))
189 self.assertEqual(5, get_tensor(self.db, sum_id, 1))
190 self.assertEqual(4, get_tensor(self.db, x_id, 2))
191 self.assertEqual(5, get_tensor(self.db, y_id, 2))
192 self.assertEqual(9, get_tensor(self.db, sum_id, 2))
228 def get_tensor(db, tag_id, step): function
|
| /external/tensorflow/tensorflow/python/tools/ |
| inspect_checkpoint.py | 54 print(reader.get_tensor(key))
59 print(reader.get_tensor(tensor_name))
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| saver.py | 91 ckpt_var_cache[k] = reader.get_tensor(k)
|
| network.py | [all...] |
| /external/tensorflow/tensorflow/contrib/rnn/python/tools/ |
| checkpoint_convert.py | 230 tensor = reader.get_tensor(var_name)
|
| /external/tensorflow/tensorflow/python/util/ |
| py_checkpoint_reader.i | 152 def get_tensor(self, tensor_str):
|
| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| checkpoint_utils.py | 80 return reader.get_tensor(name)
|
| /external/tensorflow/tensorflow/python/training/ |
| checkpoint_utils.py | 80 return reader.get_tensor(name)
|
| saver_test.py | [all...] |
| /external/tensorflow/tensorflow/python/debug/lib/ |
| stepper.py | 541 return self._dumped_intermediate_tensors[target_name].get_tensor()
633 feeds[inp] = self._dumped_intermediate_tensors[inp.name].get_tensor()
[all...] |
| session_debug_testlib.py | [all...] |
| debug_data.py | 281 loaded (with the `get_tensor` method) if needed.
342 def get_tensor(self): member in class:DebugTensorDatum
[all...] |
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| variables.py | 613 ckpt_value = reader.get_tensor(ckpt_name)
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| analyzer_cli.py | [all...] |
| /external/tensorflow/tensorflow/contrib/tensorrt/convert/ |
| convert_nodes.cc | 400 TRT_TensorOrWeights get_tensor(string name) { function in class:tensorflow::tensorrt::convert::__anon39442::Converter
[all...] |
