Searched defs:input_dim_size (Results 1 – 7 of 7) sorted by relevance
100 int input_dim_size, int offset) { in GetInputDimension()
2558 const int64 input_dim_size = operand_shape.dimensions(dim); in InferDynamicSliceShape() local2675 const int64 input_dim_size = operand_shape.dimensions(dim); in InferDynamicUpdateSliceShape() local
74 kernel_spatial_dim_size, input_dim_size; member2604 const int64 input_dim_size = in GetConvolutionDetails() local
986 const int64 input_dim_size = in HandleReshape() local
1439 int64_t input_dim_size = in Verify() local1469 int64_t input_dim_size = in Verify() local
2926 int64_t input_dim_size = input_type.getDimSize(dim_index); in matchAndRewrite() local3041 int64_t input_dim_size = input_type.getDimSize(dim_index); in matchAndRewrite() local
2043 int64 input_dim_size = operand.base_shape().dimensions(input_sharded_dim); in HandleReshape() local