Searched defs:shape_src (Results 1 – 1 of 1) sorted by relevance
56 void setup(TensorShape shape_src, int axis, DataType data_type, int num_tensors) in setup()69 TensorType compute_target(TensorShape shape_src, int axis, DataType data_type, int num_tensors) in compute_target()113 …SimpleTensor<T> compute_reference(const TensorShape &shape_src, int axis, DataType data_type, int … in compute_reference()