Searched defs:shaped (Results 1 – 5 of 5) sorted by relevance
76 if (auto shaped = type.dyn_cast<ShapedType>()) in isIllegalType() local83 if (auto shaped = type.dyn_cast<ShapedType>()) { in replaceType() local
840 typename TTypes<T, NDIMS>::Tensor Tensor::shaped( in shaped() function867 typename TTypes<T, NDIMS>::ConstTensor Tensor::shaped( in shaped() function
402 T shaped = (t.*Func)(sizes); in TestReshape() local408 T shaped = (static_cast<const Tensor&>(t).*Func)(sizes); in TestReshape() local413 void TestReshapeImpl(T shaped, std::initializer_list<int64> sizes) { in TestReshapeImpl()
191 if (auto shaped = input_type.dyn_cast<ShapedType>()) { in SetInputNodesQuantizationParams() local
208 auto shaped = target.dyn_cast<RankedTensorType>(); in ResetAxisAndBroadcast() local