Searched defs:num_dimensions (Results 1 – 8 of 8) sorted by relevance
28 def add_leading_unit_dimensions(x, num_dimensions): argument
36 int num_dimensions) { in MultiplyShapeDims()51 const int num_dimensions = NumDimensions(input); in ResizeOutput() local
80 const int num_dimensions = in Prepare() local
80 int num_dimensions = params->num_dimensions; in GetOutputShapeFromParam() local
41 const int num_dimensions = NumDimensions(input); in ResizeOutput() local
139 const int num_dimensions = window.dimensions_size(); in RunCudnnConvImpl() local
39 int num_dimensions = flat_vector->Length(); in FlatBufferIntVectorToArray() local
251 int num_dimensions; member