Lines Matching defs:shape

127     TensorType compute_target(const TensorShape &shape, DataType data_type, bool in_place)  in compute_target()
179     SimpleTensor<T> compute_reference(const TensorShape &shape, DataType data_type)  in compute_reference()
201     void setup(const TensorShape &shape, DataType data_type)  in setup()
212     void setup(const TensorShape &shape, DataType data_type)  in setup()
223     void setup(const TensorShape &shape, DataType data_type)  in setup()
234     void setup(const TensorShape &shape, DataType data_type)  in setup()
245     void setup(const TensorShape &shape, DataType data_type, bool in_place)  in setup()
256     void setup(const TensorShape &shape, DataType data_type)  in setup()
267     void setup(const TensorShape &shape, DataType data_type)  in setup()
278     void setup(const TensorShape &shape, DataType data_type)  in setup()
289     void setup(const TensorShape &shape, DataType data_type)  in setup()