Searched defs:input_dim_size (Results 1 – 7 of 7) sorted by relevance
100 int input_dim_size, int offset) { in GetInputDimension()
2672 const int64_t input_dim_size = operand_shape.dimensions(dim); in InferDynamicSliceShape() local2789 const int64_t input_dim_size = operand_shape.dimensions(dim); in InferDynamicUpdateSliceShape() local
76 kernel_spatial_dim_size, input_dim_size; member3069 const int64_t input_dim_size = in GetConvolutionDetails() local
1021 const int64_t input_dim_size = in HandleReshape() local
1494 int64_t input_dim_size = in Verify() local1524 int64_t input_dim_size = in Verify() local
3534 int64_t input_dim_size = input_type.getDimSize(dim_index); in matchAndRewrite() local3603 Value input_dim_size = in matchAndRewrite() local3725 int64_t input_dim_size = input_type.getDimSize(dim_index); in matchAndRewrite() local
2005 int64_t input_dim_size = operand.base_shape().dimensions(input_sharded_dim); in HandleReshape() local