Searched refs:ext_input_shape (Results 1 – 2 of 2) sorted by relevance
42 const RuntimeShape ext_input_shape = in PadImpl() local
4626 const RuntimeShape ext_input_shape = in PadImpl() local4666 const int input_depth = ext_input_shape.Dims(4); in PadImpl()4707 Offset(ext_input_shape, out_b - left_b_padding, in PadImpl()4802 const RuntimeShape ext_input_shape = in PadImageStyleMemset() local4828 const int batch = MatchingDim(ext_input_shape, 0, ext_output_shape, 0); in PadImageStyleMemset()4831 const int input_height = ext_input_shape.Dims(1); in PadImageStyleMemset()4832 const int input_width = ext_input_shape.Dims(2); in PadImageStyleMemset()4833 const int depth = MatchingDim(ext_input_shape, 3, ext_output_shape, 3); in PadImageStyleMemset()
Completed in 25 milliseconds