Home
last modified time | relevance | path

Searched defs:op_inputs (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/generators/
Dadd.cc57 const int32_t op_inputs[2] = {0, 1}; in Build() local
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dleaky_relu_tester.cc121 const std::array<int32_t, 1> op_inputs{{0}}; in CreateTfLiteModel() local
Dsoftmax_tester.cc123 const std::array<int32_t, 1> op_inputs{{0}}; in CreateTfLiteModel() local
Dreshape_tester.cc136 std::vector<int32_t> op_inputs({0}); in CreateTfLiteModel() local
Ddepth_to_space_tester.cc139 const std::array<int32_t, 1> op_inputs{{0}}; in CreateTfLiteModel() local
Dunary_elementwise_tester.cc149 const std::array<int32_t, 1> op_inputs{{0}}; in CreateTfLiteModel() local
Dreduce_tester.cc138 const std::array<int32_t, 2> op_inputs{{0, 1}}; in CreateTfLiteModel() local
Dresize_bilinear_tester.cc155 const std::array<int32_t, 2> op_inputs{{0, 1}}; in CreateTfLiteModel() local
Dpad_tester.cc155 const std::array<int32_t, 2> op_inputs{{0, 1}}; in CreateTfLiteModel() local
Dprelu_tester.cc199 const std::array<int32_t, 2> op_inputs{ in CreateTfLiteModel() local
Dpool_2d_tester.cc167 const std::array<int32_t, 1> op_inputs{{0}}; in CreateTfLiteModel() local
Dfully_connected_tester.cc247 const std::array<int32_t, 3> op_inputs{ in CreateTfLiteModel() local
Dbinary_elementwise_tester.cc366 const std::array<int32_t, 2> op_inputs{ in CreateTfLiteModel() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute_node.cc65 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& op_inputs, in Init()
/external/XNNPACK/bench/
Dbankers-rounding.cc133 const std::array<int32_t, 1> op_inputs{{ 0 }}; in tflite_bankers_rounding_f32() local
Delu.cc133 const std::array<int32_t, 1> op_inputs{{ 0 }}; in tflite_elu_f32() local
Dceiling.cc133 const std::array<int32_t, 1> op_inputs{{ 0 }}; in tflite_ceiling_f32() local
Dsquare-root.cc133 const std::array<int32_t, 1> op_inputs{{ 0 }}; in tflite_square_root_f32() local
Dfloor.cc133 const std::array<int32_t, 1> op_inputs{{ 0 }}; in tflite_floor_f32() local
Dprelu.cc154 const int32_t op_inputs[2] = { 0, 1 }; in tflite_prelu_f32() local
Dsoftmax.cc204 const int32_t op_inputs[1] = { 0 }; in tflite_softmax_f32() local
Dsigmoid.cc208 const std::array<int32_t, 1> op_inputs{{ 0 }}; in tflite_sigmoid_f32() local
Dhardswish.cc135 const std::array<int32_t, 1> op_inputs{{ 0 }}; in tflite_hardswish_f32() local
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_execute_op.cc87 absl::Span<const RefPtr<XRTTupleAllocation>> op_inputs, in GetChainedOpInputs()
430 -> xla::StatusOr<RefPtr<XRTTupleAllocation>> { in DoWork()
Dxrt_execute_op.cc103 absl::Span<const RefPtr<XRTTupleAllocation>> op_inputs) { in GetChainedOpInputTuples()
560 -> xla::StatusOr<RefPtr<XRTTupleAllocation>> { in DoWork()

12