Searched defs:input_dim_size (Results 1 – 7 of 7) sorted by relevance
100 int input_dim_size, int offset) { in GetInputDimension()
2675 const int64_t input_dim_size = operand_shape.dimensions(dim); in InferDynamicSliceShape() local2792 const int64_t input_dim_size = operand_shape.dimensions(dim); in InferDynamicUpdateSliceShape() local
75 low_padding_for_conv, kernel_spatial_dim_size, input_dim_size; member3566 const int64_t input_dim_size = in GetConvolutionDetails() local
1159 const int64_t input_dim_size = in HandleReshape() local
1551 int64_t input_dim_size = in verify() local1584 int64_t input_dim_size = in verify() local
3331 int64_t input_dim_size = input_type.getDimSize(dim_index); in matchAndRewrite() local3400 Value input_dim_size = in matchAndRewrite() local3533 int64_t input_dim_size = input_type.getDimSize(dim_index); in matchAndRewrite() local
2700 int64_t input_dim_size = operand.base_shape().dimensions(input_sharded_dim); in HandleReshape() local