Searched refs:ToNeonDims (Results 1 – 1 of 1) sorted by relevance
126 auto input_neon_dims = ToNeonDims(input.shape()); in Compute()138 output_ptr, ToNeonDims(out_shape)); in Compute()152 neon::Dims<4> ToNeonDims(const TensorShape& input) { in ToNeonDims() function in tensorflow::NeonDepthwiseConv2dNativeOp