Searched defs:shaped (Results 1 – 5 of 5) sorted by relevance
80 if (auto shaped = type.dyn_cast<ShapedType>()) { in ToLegalType() local
841 typename TTypes<T, NDIMS>::Tensor Tensor::shaped( in shaped() function868 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()
201 if (auto shaped = input_type.dyn_cast<ShapedType>()) { in SetInputNodesQuantizationParams() local
228 auto shaped = target.dyn_cast<RankedTensorType>(); in ResetAxisAndBroadcast() local