Searched refs:graph_inputs (Results 1 – 15 of 15) sorted by relevance
| /external/XNNPACK/bench/ |
| D | elu.cc | 141 const std::array<int32_t, 1> graph_inputs{{ 0 }}; in tflite_elu_f32() local
146 builder.CreateVector<int32_t>(graph_inputs.data(), graph_inputs.size()), in tflite_elu_f32()
|
| D | bankers-rounding.cc | 141 const std::array<int32_t, 1> graph_inputs{{ 0 }}; in tflite_bankers_rounding_f32() local
146 builder.CreateVector<int32_t>(graph_inputs.data(), graph_inputs.size()), in tflite_bankers_rounding_f32()
|
| D | floor.cc | 141 const std::array<int32_t, 1> graph_inputs{{ 0 }}; in tflite_floor_f32() local
146 builder.CreateVector<int32_t>(graph_inputs.data(), graph_inputs.size()), in tflite_floor_f32()
|
| D | square-root.cc | 141 const std::array<int32_t, 1> graph_inputs{{ 0 }}; in tflite_square_root_f32() local
146 builder.CreateVector<int32_t>(graph_inputs.data(), graph_inputs.size()), in tflite_square_root_f32()
|
| D | ceiling.cc | 141 const std::array<int32_t, 1> graph_inputs{{ 0 }}; in tflite_ceiling_f32() local
146 builder.CreateVector<int32_t>(graph_inputs.data(), graph_inputs.size()), in tflite_ceiling_f32()
|
| D | hardswish.cc | 143 const std::array<int32_t, 1> graph_inputs{{ 0 }}; in tflite_hardswish_f32() local
148 builder.CreateVector<int32_t>(graph_inputs.data(), graph_inputs.size()), in tflite_hardswish_f32()
|
| D | sigmoid.cc | 216 const std::array<int32_t, 1> graph_inputs{{ 0 }}; in tflite_sigmoid_f32() local
221 builder.CreateVector<int32_t>(graph_inputs.data(), graph_inputs.size()), in tflite_sigmoid_f32()
|
| D | prelu.cc | 162 const int32_t graph_inputs[1] = { 0 }; in tflite_prelu_f32() local
167 builder.CreateVector<int32_t>(graph_inputs, 1), in tflite_prelu_f32()
|
| D | softmax.cc | 213 const int32_t graph_inputs[1] = { 0 }; in tflite_softmax_f32() local
218 builder.CreateVector<int32_t>(graph_inputs, 1), in tflite_softmax_f32()
|
| D | average-pooling.cc | 264 const int32_t graph_inputs[1] = { 0 }; in tflite_average_pooling_f32() local
269 builder.CreateVector<int32_t>(graph_inputs, 1), in tflite_average_pooling_f32()
|
| D | deconvolution.cc | 379 const int32_t graph_inputs[1] = { 2 }; in tflite_deconvolution_f32() local
384 builder.CreateVector<int32_t>(graph_inputs, 1), in tflite_deconvolution_f32()
|
| D | convolution.cc | 653 const int32_t graph_inputs[1] = { 0 }; in tflite_convolution_f32() local
658 builder.CreateVector<int32_t>(graph_inputs, 1), in tflite_convolution_f32()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | op_selector.py | 324 def graph_inputs(op): function
359 for inp in graph_inputs(op):
414 for inp in graph_inputs(op):
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | op_callbacks_test.py | 101 self.graph_inputs.append(inputs)
148 self.graph_inputs = []
255 self.assertFalse(instrument.graph_inputs)
311 self.assertFalse(instrument.graph_inputs)
477 len(instrument.graph_inputs), len(instrument.graph_op_types))
478 unique_inputs = instrument.graph_inputs[instrument.graph_op_types.index(
485 add_inputs = instrument.graph_inputs[instrument.graph_op_types.index(
493 log_inputs = instrument.graph_inputs[instrument.graph_op_types.index(
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | lift_to_graph.py | 46 for i in op_selector.graph_inputs(_as_operation(op_or_tensor)))
278 for inp in op_selector.graph_inputs(op):
|