Home
last modified time | relevance | path

Searched defs:ArgMinMaxTestParams (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes_test.cc8448 struct ArgMinMaxTestParams { struct
8449 std::vector<int> input_shape;
8450 std::vector<float> input_value;
8451 int axis;
8452 std::vector<int> expected_output_dims;
8453 std::vector<int> expected_argmax_output;
8454 std::vector<int> expected_argmin_output;
8455 Status status;