Lines Matching defs:input_shape

59 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape…  in setup()
109 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_sha…  in compute_target()
166 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape…  in compute_reference()
247 …void setup(TensorShape input_shape, WinogradInfo winograd_info, DataLayout data_layout, DataType d…  in setup()
297 …TensorType compute_target(TensorShape input_shape, const TensorShape &output_shape, const Winograd…  in compute_target()
338 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape,…  in compute_reference()
359 …void setup(TensorShape input_shape, Size2D output_tile, DataLayout data_layout, DataType data_type)  in setup()
411 …TensorType compute_target(TensorShape input_shape, const TensorShape &output_shape, const Winograd…  in compute_target()
453 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape,…  in compute_reference()
522 …TensorType compute_target(const TensorShape &input_shape, const WinogradInfo &winograd_info, DataT…  in compute_target()
566 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, WinogradInfo winograd_info, Data…  in compute_reference()