Home
last modified time | relevance | path

Searched refs:inputs_data (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/
Dconcatenation.cc91 const data_type* inputs_data[kMaxInputNum]; in EvalUnquantized() local
94 GetAllInputTensorData(context, node, inputs_data); in EvalUnquantized()
102 reference_ops::Concatenation(data->params, inputs_shape_ptr, inputs_data, in EvalUnquantized()
111 const uint8_t* inputs_data[kMaxInputNum]; in EvalQuantizedUInt8() local
114 GetAllInputTensorData(context, node, inputs_data); in EvalQuantizedUInt8()
123 data->params, inputs_shape_ptr, inputs_data, in EvalQuantizedUInt8()
Dreshape_test.cc79 int inputs_data[] = {2, 0, 1}; in TestReshapeWithShape() local
80 TfLiteIntArray* inputs_array = IntArrayFromInts(inputs_data); in TestReshapeWithShape()
105 int inputs_data[] = {1, 0}; in TestReshapeWithoutShape() local
106 TfLiteIntArray* inputs_array = IntArrayFromInts(inputs_data); in TestReshapeWithoutShape()
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py363 def _GetFeedDict(self, inputs_data): argument
364 return {name: data for name, data in zip(self._GetFeedNames(), inputs_data)}
366 def _RunGraphV1(self, saved_model_dir, inputs_data, config, num_runs=2): argument
377 params.expected_output_dims, inputs_data):
392 def _RunGraphV2(self, saved_model_dir, inputs_data, graph_state, num_runs=2): argument
400 inputs_data):
429 inputs_data, argument
433 for data in inputs_data:
437 results = self._RunGraphV2(saved_model_dir, inputs_data, graph_state,
449 return self._RunGraphV1(saved_model_dir, inputs_data, config, num_runs)
[all …]
/external/tensorflow/tensorflow/python/distribute/
Dcross_device_ops_test.py340 inputs_data = [1.0, 2.0, 3.0, 4.0]
341 inputs = inputs_data[0:group_size]
383 inputs_data = [
393 inputs = inputs_data[0:group_size]
464 inputs_data = [[1.0, 2.0], [3.0, 4.0], [5.0, 6.0], [7.0, 8.0]]
465 inputs = inputs_data[0:group_size]
509 inputs_data = ([
530 inputs = inputs_data[0:group_size]