Searched defs:shape_w (Results 1 – 1 of 1) sorted by relevance
51 …void setup(TensorShape shape_w, DataType data_type, DataLayout data_layout, bool in_place, bool wi… in setup()64 …std::pair<TensorType, TensorType> compute_target(TensorShape shape_w, DataType data_type, DataLayo… in compute_target()141 …std::pair<SimpleTensor<T>, SimpleTensor<T>> compute_reference(TensorShape shape_w, DataType data_t… in compute_reference()