Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/training/
Dpy_checkpoint_reader.py66 def get_tensor(self, tensor_str): function
77 CheckpointReader.get_tensor = get_tensor
Dcheckpoint_utils.py85 return reader.get_tensor(name)
/external/tensorflow/tensorflow/lite/python/
Dinterpreter_test.py129 output_data = interpreter.get_tensor(output_details[0]['index'])
146 output_data = interpreter.get_tensor(output_details[0]['index'])
183 output_data = interpreter.get_tensor(output_details[0]['index'])
224 output_data = interpreter.get_tensor(output_details[0]['index'])
236 test_input_tensor = interpreter.get_tensor(input_details[0]['index'])
266 compressed_buffer = interpreter.get_tensor(0)
341 self.interpreter.get_tensor(self.input0), self.initial_data)
346 array_initial_copy = self.interpreter.get_tensor(self.input0)
350 self.assertAllEqual(self.interpreter.get_tensor(self.input0), new_value)
Dlite_flex_test.py75 output_data = interpreter.get_tensor(output_details[0]['index'])
107 output_data = interpreter.get_tensor(output_details[0]['index'])
142 output_data = interpreter.get_tensor(output_details[0]['index'])
224 output_data = interpreter.get_tensor(output_details[0]['index'])
Dlite_v2_test_util.py67 interpreter.get_tensor(details['index']) for details in output_details
Dinterpreter.py230 result[output_name] = self._interpreter.get_tensor(output_index)
682 def get_tensor(self, tensor_index): member in Interpreter
Dutil_test.py137 output_data = interpreter.get_tensor(output_details["index"])[0]
384 output_data = interpreter.get_tensor(output_details["index"])[0]
Dlite_test.py512 output_data = interpreter.get_tensor(output_details[0]['index'])
1375 self.assertEqual(1.0, interpreter.get_tensor(output_details[0]['index']))
1376 self.assertEqual(2.0, interpreter.get_tensor(output_details[1]['index']))
1377 self.assertEqual(3.0, interpreter.get_tensor(output_details[2]['index']))
1378 self.assertEqual(4.0, interpreter.get_tensor(output_details[3]['index']))
1539 output_data = interpreter.get_tensor(output_details[0]['index'])
2239 tflite_result = interpreter.get_tensor(output_details[0]['index'])
2269 tflite_result = interpreter.get_tensor(output_details[0]['index'])
2391 tflite_result = interpreter.get_tensor(output_details[0]['index'])
2596 tflite_result = interpreter.get_tensor(output_details[0]['index'])
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
Dngrams_test.py125 tflite_output = interpreter.get_tensor(
145 tflite_output = interpreter.get_tensor(
170 tflite_output_values = interpreter.get_tensor(
175 tflite_output_cur_row_splits = interpreter.get_tensor(
201 tflite_output_values = interpreter.get_tensor(
206 tflite_output_cur_row_splits = interpreter.get_tensor(
Dwhitespace_tokenizer_test.py83 tflite_output = interpreter.get_tensor(
110 tflite_output_cur_row_splits = interpreter.get_tensor(
116 tflite_output_values = interpreter.get_tensor(
/external/tflite-support/tensorflow_lite_support/custom_ops/python/
Dsentencepiece_tokenizer_test.py144 interpreter.get_tensor(output_details[0]["index"]), expected_result)
186 interpreter.get_tensor(output_details[0]["index"]), expected_result)
221 interpreter.get_tensor(output_details[0]["index"]), expected_result)
Dragged_tensor_to_tensor_test.py52 interpreter.get_tensor(output_details[0]["index"]),
/external/tensorflow/tensorflow/python/tools/
Dinspect_checkpoint.py82 print(reader.get_tensor(key))
95 print(reader.get_tensor(tensor_name))
/external/tensorflow/tensorflow/python/kernel_tests/signal/
Dtest_util.py99 return [the_interpreter.get_tensor(details['index'])
/external/tensorflow/tensorflow/lite/tutorials/
Dmnist_tflite.py67 output_data = interpreter.get_tensor(output_details[0]['index'])
/external/tensorflow/tensorflow/lite/experimental/quantization_debugger/
Ddebugger.py174 diffs = self._quant_interpreter.get_tensor(tensor_details['index'])
276 tensor = interpreter.get_tensor(output_detail['index'])
/external/tensorflow/tensorflow/lite/examples/python/
Dlabel_image.py90 output_data = interpreter.get_tensor(output_details[0]['index'])
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.lite.-interpreter.pbtxt30 name: "get_tensor"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.lite.-interpreter.pbtxt30 name: "get_tensor"
/external/tensorflow/tensorflow/python/debug/lib/
Dsession_debug_testlib.py1034 self.assertAllClose([[5.0, 3.0], [-1.0, 0.0]], datum.get_tensor())
1064 self.assertAllClose(8.0, update_u_data[0].get_tensor())
1073 self.assertAllClose(19.0, update_v_data[0].get_tensor())
1120 self.assertAllClose([2, 3, 5], unique_x_slot_0_dumps[0].get_tensor())
1130 unique_x_slot_1_dumps[0].get_tensor())
1219 self.assertAllClose(np.array([[1.0, 2.0]]), m_dumps[0].get_tensor())
1223 self.assertAllClose(np.array([[-3.0, 0.0]]), x_dumps[0].get_tensor())
Ddebug_data.py343 def get_tensor(self): member in DebugTensorDatum
1458 if predicate(datum, datum.get_tensor()):
1525 return [datum.get_tensor() for datum in
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Dbidirectional_sequence_rnn_test.py277 result = interpreter.get_tensor(output_index)
Dunidirectional_sequence_lstm_test.py237 result = interpreter.get_tensor(output_index)
Dunidirectional_sequence_rnn_test.py228 result = interpreter.get_tensor(output_index)
Dbidirectional_sequence_lstm_test.py252 result = interpreter.get_tensor(output_index)

12