Searched defs:indicesShape (Results 1 – 7 of 7) sorted by relevance
23 std::vector<int32_t> indicesShape{ 3, 1 }; in GatherNdUint8Test() local43 std::vector<int32_t> indicesShape{ 3, 1 }; in GatherNdFp32Test() local
23 std::vector<int32_t> indicesShape{3}; in GatherUint8Test() local45 std::vector<int32_t> indicesShape{3}; in GatherFp32Test() local
26 std::vector<int32_t>& indicesShape, in CreateGatherNdTfLiteModel()121 std::vector<int32_t>& indicesShape, in GatherNdTest()
26 std::vector<int32_t>& indicesShape, in CreateGatherTfLiteModel()122 std::vector<int32_t>& indicesShape, in GatherTest()
32 const int32_t* indicesData, const Shape& indicesShape, T* outputData) { in eval()
308 std::vector<unsigned int> indicesShape; in CalculateGatherNdKeyIndices() local
1976 const TensorShape& indicesShape = m_TensorsInfo[node.input(1)].m_info->GetShape(); in ParseGather() local