Home
last modified time | relevance | path

Searched refs:get_tensor (Results 1 – 25 of 40) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/quantize/python/
Dgraph_matcher_test.py79 self.assertEqual(match_result.get_tensor(inputs_pattern), inputs)
80 self.assertEqual(match_result.get_tensor('inputs'), inputs)
107 self.assertEqual(match_result.get_tensor(y1_pattern), y1)
152 self.assertEqual(match_result.get_tensor(reshape_pattern), reshape)
153 self.assertEqual(match_result.get_tensor('slice'), None)
163 self.assertEqual(match_result.get_tensor(reshape_pattern), reshape)
164 self.assertEqual(match_result.get_tensor('slice'), slicing)
Dquantize.py486 weight_tensor = match_result.get_tensor(weight_identity_pattern)
488 weight_tensor = match_result.get_tensor(weight_resource_var_pattern)
490 weight_tensor = match_result.get_tensor(folded_weight_pattern)
509 weight_tensor = match_result.get_tensor(weight_identity_pattern)
511 weight_tensor = match_result.get_tensor(weight_resource_var_pattern)
513 weight_tensor = match_result.get_tensor(folded_weight_pattern)
534 weight_tensor = match_result.get_tensor(weight_identity_pattern)
536 weight_tensor = match_result.get_tensor(weight_resource_var_pattern)
538 weight_tensor = match_result.get_tensor(folded_weight_pattern)
551 weight_tensor = match_result.get_tensor(weight_identity_pattern)
[all …]
Dfold_batch_norms.py232 layer_tensor = match_result.get_tensor(layer_pattern)
257 input_tensor = match_result.get_tensor(input_pattern)
258 weight_tensor = match_result.get_tensor(weight_pattern)
259 gamma_tensor = match_result.get_tensor(gamma_pattern)
260 beta_tensor = match_result.get_tensor(beta_pattern)
297 bn_decay_mean_tensor = mul_match_result.get_tensor(bn_decay_pattern)
301 bn_decay_var_tensor = mul_match_result.get_tensor(bn_decay_pattern)
303 mean_tensor = match_result.get_tensor(mean_pattern)
304 variance_tensor = match_result.get_tensor(variance_pattern)
Dgraph_matcher.py190 def get_tensor(self, pattern_or_name): member in MatchResult
/external/tensorflow/tensorflow/lite/python/
Dinterpreter_test.py57 output_data = interpreter.get_tensor(output_details[0]['index'])
91 output_data = interpreter.get_tensor(output_details[0]['index'])
126 output_data = interpreter.get_tensor(output_details[0]['index'])
168 self.interpreter.get_tensor(self.input0), self.initial_data)
173 array_initial_copy = self.interpreter.get_tensor(self.input0)
177 self.assertAllEqual(self.interpreter.get_tensor(self.input0), new_value)
Dlite_test.py290 output_data = interpreter.get_tensor(output_details[0]['index'])
624 self.assertEqual(1.0, interpreter.get_tensor(output_details[0]['index']))
625 self.assertEqual(2.0, interpreter.get_tensor(output_details[1]['index']))
626 self.assertEqual(3.0, interpreter.get_tensor(output_details[2]['index']))
627 self.assertEqual(4.0, interpreter.get_tensor(output_details[3]['index']))
1082 tflite_result = interpreter.get_tensor(output_details[0]['index'])
1204 tflite_result = interpreter.get_tensor(output_details[0]['index'])
1330 tflite_result = interpreter.get_tensor(output_details[0]['index'])
Dinterpreter.py216 def get_tensor(self, tensor_index): member in Interpreter
Dlite_v2_test.py51 return interpreter.get_tensor(output_details[0]['index'])
/external/tensorflow/tensorflow/contrib/summary/
Dsummary_ops_test.py348 self.assertEqual(2, get_tensor(self.db, x_id, 1))
349 self.assertEqual(3, get_tensor(self.db, y_id, 1))
350 self.assertEqual(5, get_tensor(self.db, sum_id, 1))
351 self.assertEqual(4, get_tensor(self.db, x_id, 2))
352 self.assertEqual(5, get_tensor(self.db, y_id, 2))
353 self.assertEqual(9, get_tensor(self.db, sum_id, 2))
389 def get_tensor(db, tag_id, step): function
/external/tensorflow/tensorflow/python/tools/
Dinspect_checkpoint.py54 print(reader.get_tensor(key))
59 print(reader.get_tensor(tensor_name))
/external/tensorflow/tensorflow/lite/tutorials/
Dmnist_tflite.py66 output_data = interpreter.get_tensor(output_details[0]['index'])
/external/tensorflow/tensorflow/lite/examples/python/
Dlabel_image.py77 output_data = interpreter.get_tensor(output_details[0]['index'])
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.lite.-interpreter.pbtxt22 name: "get_tensor"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.lite.-interpreter.pbtxt22 name: "get_tensor"
/external/tensorflow/tensorflow/contrib/eager/python/
Dsaver.py91 ckpt_var_cache[k] = reader.get_tensor(k)
/external/tensorflow/tensorflow/python/debug/lib/
Dstepper.py541 return self._dumped_intermediate_tensors[target_name].get_tensor()
633 feeds[inp] = self._dumped_intermediate_tensors[inp.name].get_tensor()
940 return self._dumped_intermediate_tensors[tensor_name].get_tensor()
Dsession_debug_testlib.py1036 self.assertAllClose([[5.0, 3.0], [-1.0, 0.0]], datum.get_tensor())
1066 self.assertAllClose(8.0, update_u_data[0].get_tensor())
1075 self.assertAllClose(19.0, update_v_data[0].get_tensor())
1122 self.assertAllClose([2, 3, 5], unique_x_slot_0_dumps[0].get_tensor())
1132 unique_x_slot_1_dumps[0].get_tensor())
1221 self.assertAllClose(np.array([[1.0, 2.0]]), m_dumps[0].get_tensor())
1225 self.assertAllClose(np.array([[-3.0, 0.0]]), x_dumps[0].get_tensor())
Ddebug_data.py343 def get_tensor(self): member in DebugTensorDatum
1457 if predicate(datum, datum.get_tensor()):
1524 return [datum.get_tensor() for datum in
/external/tensorflow/tensorflow/contrib/tensor_forest/client/
Drandom_forest_test.py144 reader.get_tensor(ops.GraphKeys.GLOBAL_STEP), global_step)
330 reader.get_tensor(ops.GraphKeys.GLOBAL_STEP), global_step)
/external/tensorflow/tensorflow/contrib/rnn/python/tools/
Dcheckpoint_convert.py230 tensor = reader.get_tensor(var_name)
/external/tensorflow/tensorflow/contrib/framework/python/framework/
Dcheckpoint_utils.py80 return reader.get_tensor(name)
/external/tensorflow/tensorflow/python/util/
Dpy_checkpoint_reader.i152 def get_tensor(self, tensor_str):
/external/tensorflow/tensorflow/python/training/tracking/
Dutil.py150 [reader.get_tensor(name) for name in spec_names])
323 object_graph_string = reader.get_tensor(
1058 object_graph_string = reader.get_tensor(
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_utils.py82 return reader.get_tensor(name)
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Dunidirectional_sequence_rnn_test.py178 result = interpreter.get_tensor(output_index)

12