Searched defs:paramsShape (Results 1 – 9 of 9) sorted by relevance
22 std::vector<int32_t> paramsShape{ 5, 2 }; in GatherNdUint8Test() local42 std::vector<int32_t> paramsShape{ 5, 2 }; in GatherNdFp32Test() local
22 std::vector<int32_t> paramsShape{8}; in GatherUint8Test() local44 std::vector<int32_t> paramsShape{8}; in GatherFp32Test() local
25 std::vector<int32_t>& paramsShape, in CreateGatherNdTfLiteModel()120 std::vector<int32_t>& paramsShape, in GatherNdTest()
25 std::vector<int32_t>& paramsShape, in CreateGatherTfLiteModel()121 std::vector<int32_t>& paramsShape, in GatherTest()
29 const TensorShape& paramsShape = paramsInfo.GetShape(); in Gather() local
47 TensorShape paramsShape = inputInfo0.GetShape(); in Execute() local
128 TensorShape paramsShape = paramsInfo.GetShape(); in NeonGatherNdWorkload() local
131 TensorShape paramsShape = paramsInfo.GetShape(); in ClGatherNdWorkload() local
302 std::vector<unsigned int> paramsShape; in CalculateGatherNdKeyIndices() local