Searched defs:num_dimensions (Results 1 – 15 of 15) sorted by relevance
33 def add_leading_unit_dimensions(x, num_dimensions): # pylint: disable=invalid-name argument
41 int num_dimensions) { in MultiplyShapeDims()61 const int num_dimensions = NumDimensions(input); in ResizeOutput() local
90 const int num_dimensions = in Prepare() local
92 int num_dimensions = params->num_dimensions; in GetOutputShapeFromParam() local
48 const int num_dimensions = NumDimensions(input); in ResizeOutput() local
41 int num_dimensions = params->num_dimensions; in PopulateShapeFromParam() local
164 int num_dimensions = NumDimensions(input); in Prepare() local
222 const int num_dimensions = output_tensor->dims->size; in CheckAndSetOutputs() local
330 const int num_dimensions = window.dimensions_size(); in GetGpuConvConfig() local
2543 ArrayRef<int64_t> output_shape, int num_dimensions, in ComputePermutation()2579 const int num_dimensions = input_tensor.getType().getRank(); in fold() local
334 int num_dimensions; member
1913 int num_dimensions = params->num_dimensions; in Validate() local2861 int num_dimensions = params->num_dimensions; in Map() local
88 size_t num_dimensions = flat_vector->size(); in FlatBufferIntVectorToArray() local
964 const int64 num_dimensions = lengths.size(); in ComputeStrides() local